m5/UNITV_TRAIN
0
0
Fork 0
mirror of https://github.com/m5stack/UNITV_TRAIN.git synced 2026-05-20 11:05:22 -07:00
Code Issues Packages Projects Releases Wiki Activity

update template

Browse Source
This commit is contained in:
Neucrack
2020-11-20 17:25:30 +08:00
parent 2265817dc0
commit 97438cd1b8
2 changed files with 31 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train/classifier/template/boot.py
+15 -3
View File
@@ -5,15 +5,26 @@ import sensor, image, lcd, time
import KPU as kpu
import gc, sys
def lcd_show_except(e):
import uio
err_str = uio.StringIO()
sys.print_exception(e, err_str)
err_str = err_str.getvalue()
img = image.Image(size=(224,224))
img.draw_string(0, 10, err_str, scale=1, color=(0xff,0x00,0x00))
lcd.display(img)
def main(labels = None, model_addr="/sd/m.kmodel"):
def main(labels = None, model_addr="/sd/m.kmodel", sensor_window=(224, 224), lcd_rotation=0, sensor_hmirror=False, sensor_vflip=False):
sensor.reset()
sensor.set_pixformat(sensor.RGB565)
sensor.set_framesize(sensor.QVGA)
sensor.set_windowing((224, 224))
sensor.set_windowing(sensor_window)
sensor.set_hmirror(sensor_hmirror)
sensor.set_vflip(sensor_vflip)
sensor.run(1)
lcd.init(type=1)
lcd.rotation(lcd_rotation)
lcd.clear(lcd.WHITE)
if not labels:
@@ -48,7 +59,7 @@ def main(labels = None, model_addr="/sd/m.kmodel"):
img.draw_string(0, 200, "t:%dms" %(t), scale=2)
lcd.display(img)
except Exception as e:
sys.print_exception(e)
raise e
finally:
kpu.deinit(task)
@@ -60,5 +71,6 @@ if __name__ == "__main__":
main(labels=labels, model_addr="/sd/m.kmodel")
except Exception as e:
sys.print_exception(e)
lcd_show_except(e)
finally:
gc.collect()
train/detector/template/boot.py
+16 -4
View File
@@ -5,15 +5,26 @@ import sensor, image, lcd, time
import KPU as kpu
import gc, sys
def lcd_show_except(e):
import uio
err_str = uio.StringIO()
sys.print_exception(e, err_str)
err_str = err_str.getvalue()
img = image.Image(size=(224,224))
img.draw_string(0, 10, err_str, scale=1, color=(0xff,0x00,0x00))
lcd.display(img)
def main(anchors, labels = None, model_addr="/sd/m.kmodel"):
def main(anchors, labels = None, model_addr="/sd/m.kmodel", sensor_window=(224, 224), lcd_rotation=0, sensor_hmirror=False, sensor_vflip=False):
sensor.reset()
sensor.set_pixformat(sensor.RGB565)
sensor.set_framesize(sensor.QVGA)
sensor.set_windowing((224, 224))
sensor.set_windowing(sensor_window)
sensor.set_hmirror(sensor_hmirror)
sensor.set_vflip(sensor_vflip)
sensor.run(1)
lcd.init(type=1)
lcd.rotation(lcd_rotation)
lcd.clear(lcd.WHITE)
if not labels:
@@ -49,7 +60,7 @@ def main(anchors, labels = None, model_addr="/sd/m.kmodel"):
img.draw_string(0, 200, "t:%dms" %(t), scale=2)
lcd.display(img)
except Exception as e:
sys.print_exception(e)
raise e
finally:
kpu.deinit(task)
@@ -58,9 +69,10 @@ if __name__ == "__main__":
try:
labels = [] # labels
anchors = [] # anchors
# main(anchors = anchors, labels=labels, model_addr=0x300000)
# main(anchors = anchors, labels=labels, model_addr=0x300000, lcd_rotation=0)
main(anchors = anchors, labels=labels, model_addr="/sd/m.kmodel")
except Exception as e:
sys.print_exception(e)
lcd_show_except(e)
finally:
gc.collect()