Skip to content

Commit

Permalink
add scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
knorth55 committed Aug 2, 2019
1 parent c2cab48 commit 0336d6f
Show file tree
Hide file tree
Showing 2 changed files with 47 additions and 0 deletions.
25 changes: 25 additions & 0 deletions scripts/demo_find_face.py
Original file line number Diff line number Diff line change
@@ -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)
22 changes: 22 additions & 0 deletions scripts/demo_fps_display.py
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit 0336d6f

Please sign in to comment.