Adding in training of dual model recognition system.

This commit is contained in:
Lucas Oskorep
2021-03-05 16:18:12 -06:00
parent 9822ace85a
commit 90f6b16731
7 changed files with 92 additions and 6 deletions
+4 -4
View File
@@ -6,9 +6,9 @@ import multiprocessing
train_dir = "./train/"
test_dir = "./test/"
val_dir = "./val/"
train = .80
test = .15
val = .05
train = .70
test = .30
val = .00
def add_train_data(file, filename, label):
@@ -62,7 +62,7 @@ def test_split_file(file_root):
file = file_root[1]
# print(file)
if file is ".DS_Store":
if file == ".DS_Store":
return
c = random()