Update smile script.

This commit is contained in:
iabdalkader 2018-06-22 15:14:15 +02:00
parent 1438108522
commit 3a45b3ce17

View File

@ -3,8 +3,8 @@ import sensor, time, image, os, nn
sensor.reset() # Reset and initialize the sensor.
sensor.set_contrast(2)
sensor.set_pixformat(sensor.GRAYSCALE) # Set pixel format to RGB565
sensor.set_framesize(sensor.QQVGA) # Set frame size to QVGA (320x240)
sensor.set_pixformat(sensor.RGB565) # Set pixel format to RGB565
sensor.set_framesize(sensor.QVGA) # Set frame size to QVGA (320x240)
sensor.skip_frames(time=2000)
sensor.set_auto_gain(False)
@ -29,7 +29,7 @@ while (True):
# Detect smiles
for r in objects:
# Resize and center detection area
r = [r[0], r[1]+10, int(r[2]*1.1), int(r[2]*1.1)]
r = [r[0]+10, r[1]+25, int(r[2]*0.70), int(r[2]*0.70)]
img.draw_rectangle(r)
out = net.forward(img, roi=r, softmax=True)
img.draw_string(r[0], r[1], ':)' if (out[0] > 0.8) else ':(', color=(255), scale=2)