diff --git a/scripts/demo_find_face.py b/scripts/demo_find_face.py new file mode 100644 index 0000000..9df9f76 --- /dev/null +++ b/scripts/demo_find_face.py @@ -0,0 +1,25 @@ +import KPU as kpu +import lcd +import sensor + + +def find_face(): + lcd.init() + sensor.reset() + sensor.set_pixformat(sensor.RGB565) + sensor.set_framesize(sensor.QVGA) + sensor.set_vflip(1) + sensor.run(1) + task = kpu.load(0x300000) + anchor = ( + 1.889, 2.5245, 2.9465, 3.94056, 3.99987, + 5.3658, 5.155437, 6.92275, 6.718375, 9.01025) + kpu.init_yolo2(task, 0.5, 0.3, 5, anchor) + while(True): + img = sensor.snapshot() + code = kpu.run_yolo2(task, img) + if code: + for i in code: + img.draw_rectangle(i.rect()) + lcd.display(img) + kpu.deinit(task) diff --git a/scripts/demo_fps_display.py b/scripts/demo_fps_display.py new file mode 100644 index 0000000..2413bb4 --- /dev/null +++ b/scripts/demo_fps_display.py @@ -0,0 +1,22 @@ +import lcd +import sensor +import time + + +def fps_display(): + clock = time.clock() + lcd.init() + sensor.reset() + sensor.set_pixformat(sensor.RGB565) + sensor.set_framesize(sensor.QVGA) + sensor.set_vflip(1) + sensor.run(1) + sensor.skip_frames(30) + while True: + clock.tick() + img = sensor.snapshot() + fps = clock.fps() + img.draw_string( + 2, 2, ("%2.1ffps" % fps), + color=(0, 128, 0), scale=2) + lcd.display(img)