mirror of
https://github.com/openmv/openmv.git
synced 2025-09-26 23:09:13 +08:00
Remove old cifar stuff
This commit is contained in:
parent
6d4889f1c5
commit
ab6ab896d7
Binary file not shown.
@ -1,15 +0,0 @@
|
||||
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(data_path+'/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 ((match / total * 100 ) > 90)
|
Loading…
Reference in New Issue
Block a user