import sensorimport mathimport machineimport lcdimport KPU as kpusensor.reset()sensor.set_pixformat(sensor.RGB565)sensor.set_framesize(sensor.QVGA)sensor.run(1)sensor.skip_frames(10)sensor.set_windowing((224,224))sensor.set_vflip(0)lcd.init(freq=15000000,color=0x0000)KPU = kpu.load('/sd/weights.kmodel')success = kpu.set_outputs(KPU,0,1,1,2)f=open('classes.txt','r')labels_txt=f.read()labels = labels_txt.split(',')f.close()while True: img = sensor.snapshot() plist = kpu.forward(KPU,img)[:] pmax = max(plist) max_index = plist.index(pmax) pgoods = labels[max_index] print('No.{} is {}'.format(pgoods, pmax)) lcd.display(img) lcd.draw_string(48,224,'NO: {} {}{}'.format(pgoods, round(pmax,2)*100,'%'),lcd.WHITE,lcd.GREEN)1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
301
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30