mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
Add NN unit test
This commit is contained in:
parent
9cc33a16a7
commit
9a61a73711
154
scripts/unittest/data/cifar10/airplane-1.ppm
Normal file
154
scripts/unittest/data/cifar10/airplane-1.ppm
Normal file
File diff suppressed because one or more lines are too long
7
scripts/unittest/data/cifar10/airplane-2.ppm
Normal file
7
scripts/unittest/data/cifar10/airplane-2.ppm
Normal file
File diff suppressed because one or more lines are too long
18
scripts/unittest/data/cifar10/airplane-3.ppm
Normal file
18
scripts/unittest/data/cifar10/airplane-3.ppm
Normal file
File diff suppressed because one or more lines are too long
215
scripts/unittest/data/cifar10/bird-1.ppm
Normal file
215
scripts/unittest/data/cifar10/bird-1.ppm
Normal file
File diff suppressed because one or more lines are too long
76
scripts/unittest/data/cifar10/bird-2.ppm
Normal file
76
scripts/unittest/data/cifar10/bird-2.ppm
Normal file
File diff suppressed because one or more lines are too long
284
scripts/unittest/data/cifar10/bird-3.ppm
Normal file
284
scripts/unittest/data/cifar10/bird-3.ppm
Normal file
File diff suppressed because one or more lines are too long
135
scripts/unittest/data/cifar10/cat-1.ppm
Normal file
135
scripts/unittest/data/cifar10/cat-1.ppm
Normal file
File diff suppressed because one or more lines are too long
94
scripts/unittest/data/cifar10/cat-2.ppm
Normal file
94
scripts/unittest/data/cifar10/cat-2.ppm
Normal file
File diff suppressed because one or more lines are too long
10
scripts/unittest/data/cifar10/cat-3.ppm
Normal file
10
scripts/unittest/data/cifar10/cat-3.ppm
Normal file
File diff suppressed because one or more lines are too long
636
scripts/unittest/data/cifar10/dog-1.ppm
Normal file
636
scripts/unittest/data/cifar10/dog-1.ppm
Normal file
File diff suppressed because one or more lines are too long
65
scripts/unittest/data/cifar10/dog-2.ppm
Normal file
65
scripts/unittest/data/cifar10/dog-2.ppm
Normal file
File diff suppressed because one or more lines are too long
BIN
scripts/unittest/data/cifar10/dog-3.ppm
Normal file
BIN
scripts/unittest/data/cifar10/dog-3.ppm
Normal file
Binary file not shown.
BIN
scripts/unittest/data/cifar10/frog-1.ppm
Normal file
BIN
scripts/unittest/data/cifar10/frog-1.ppm
Normal file
Binary file not shown.
479
scripts/unittest/data/cifar10/frog-2.ppm
Normal file
479
scripts/unittest/data/cifar10/frog-2.ppm
Normal file
File diff suppressed because one or more lines are too long
BIN
scripts/unittest/data/cifar10/frog-3.ppm
Normal file
BIN
scripts/unittest/data/cifar10/frog-3.ppm
Normal file
Binary file not shown.
23
scripts/unittest/data/cifar10/horse-1.ppm
Normal file
23
scripts/unittest/data/cifar10/horse-1.ppm
Normal file
File diff suppressed because one or more lines are too long
47
scripts/unittest/data/cifar10/horse-2.ppm
Normal file
47
scripts/unittest/data/cifar10/horse-2.ppm
Normal file
File diff suppressed because one or more lines are too long
194
scripts/unittest/data/cifar10/horse-3.ppm
Normal file
194
scripts/unittest/data/cifar10/horse-3.ppm
Normal file
File diff suppressed because one or more lines are too long
1528
scripts/unittest/data/cifar10/ship-1.ppm
Normal file
1528
scripts/unittest/data/cifar10/ship-1.ppm
Normal file
File diff suppressed because one or more lines are too long
158
scripts/unittest/data/cifar10/ship-2.ppm
Normal file
158
scripts/unittest/data/cifar10/ship-2.ppm
Normal file
File diff suppressed because one or more lines are too long
9
scripts/unittest/data/cifar10/ship-3.ppm
Normal file
9
scripts/unittest/data/cifar10/ship-3.ppm
Normal file
File diff suppressed because one or more lines are too long
15
scripts/unittest/script/21-nn_cifar10.py
Normal file
15
scripts/unittest/script/21-nn_cifar10.py
Normal 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)
|
||||
Loading…
Reference in New Issue
Block a user