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:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user