Add NN unit test

This commit is contained in:
iabdalkader 2018-06-21 21:59:31 +02:00
parent 9cc33a16a7
commit 9a61a73711
22 changed files with 4147 additions and 0 deletions

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

Binary file not shown.

Binary file not shown.

File diff suppressed because one or more lines are too long

Binary file not shown.

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

View File

@ -0,0 +1,15 @@
def unittest(data_path, temp_path):
import os, nn, image
match = 0
test_dir = data_path+"/cifar10"
files = os.listdir(test_dir)
total = len(files)
labels = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
net = nn.load('/cifar10_fast.network')
for f in sorted(files):
img = image.Image(test_dir+"/"+f, copy_to_fb=True)
out = net.forward(img)
label = labels[out.index(max(out))]
if (f.split('-')[0] == label):
match += 1
return (total == 21 and match == 19)