Adding in new trained models, as well as new tools for reviewing the results.

Added in testing flow for testing our unfininshed/finished models.

Also adding a test dataset with one picture of every pokemon in the game.
This commit is contained in:
Lucas Oskorep
2019-04-30 13:00:24 -05:00
parent 601f4f7633
commit fbac5d519a
17 changed files with 1643 additions and 73 deletions
+2 -3
View File
@@ -6,8 +6,8 @@ import multiprocessing
train_dir = "./data/train/"
test_dir = "./data/test/"
val_dir = "./data/val/"
train = .75
test = .20
train = .80
test = .15
val = .05
@@ -24,7 +24,6 @@ def add_train_data(file, filename, label):
print(e)
print("INVALID FILE")
os.remove(file)
# TODO: Remove the files
def add_val_data(file, filename, label):