diff --git a/usr/examples/09-Feature-Detection/find_numbers.py b/usr/examples/09-Feature-Detection/find_numbers.py index 023788ac6..5c45fed22 100644 --- a/usr/examples/09-Feature-Detection/find_numbers.py +++ b/usr/examples/09-Feature-Detection/find_numbers.py @@ -1,23 +1,20 @@ # LetNet Example import sensor, image, time -sensor.reset() # Reset and initialize the sensor. +sensor.reset() # Reset and initialize the sensor. sensor.set_contrast(3) -sensor.set_pixformat(sensor.GRAYSCALE) # Set pixel format to RGB565 (or GRAYSCALE) -sensor.set_framesize(sensor.QVGA) # Set frame size to QVGA (320x240) -sensor.set_windowing((28, 28)) - -sensor.skip_frames(time = 2000) # Wait for settings take effect. +sensor.set_pixformat(sensor.GRAYSCALE) # Set pixel format to RGB565 (or GRAYSCALE) +sensor.set_framesize(sensor.VGA) # Set frame size to QVGA (320x240) +sensor.set_windowing((128, 128)) # Set 128x128 window. +sensor.skip_frames(time = 2000) # Wait for settings take effect. sensor.set_auto_gain(False) sensor.set_auto_exposure(False) -clock = time.clock() # Create a clock object to track the FPS. - while(True): - clock.tick() # Update the FPS clock. - img = sensor.snapshot() # Take a picture and return the image. - out = img.invert().find_number() - if out[1] > 3.0: - print(out[0]) - #print(clock.fps()) # Note: OpenMV Cam runs about half as fast when connected - # to the IDE. The FPS should increase once disconnected. + img = sensor.snapshot() + # NOTE: Uncomment to detect dark numbers on white background + # img.invert() + out = img.find_number(roi=(img.width()//2-14, img.height()//2-14, 28, 28)) + img.draw_rectangle((img.width()//2-15, img.height()//2-15, 30, 30), color=(255, 255, 255)) + if out[1] > 5: # Confidence level + print("Number: %d Confidence: %0.2f" %(out[0], out[1]))