想制作目标检测,检测到目标后,控制某个管脚输出高电平,如果没有检测到,控制管脚低电平,管脚口为11,
想接继电器控制水泵开或者关。
import sensor, image, lcd, time
import KPU as kpu
import gc, sys
from fpioa_manager import fm
from Maix import GPIO
io_water_beng = 11
fm.register(io_water_beng, fm.fpioa.GPIO0)
led_r=GPIO(GPIO.GPIO0, GPIO.OUT)
led_r.value(0)
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=(320,240))
img.draw_string(0, 10, err_str, scale=1, color=(0xff,0x00,0x00))
lcd.display(img)
def main(anchors, labels = None, model_addr="/sd/yolov2.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(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:
f=open("/sd/lable.txt","r")#加载标签文件
labels_txt=f.read()
labels = labels_txt.split(",")
print(labels)
f.close()
try:
img = image.Image("startup.jpg")
lcd.display(img)
except Exception:
img = image.Image(size=(320, 240))
img.draw_string(90, 110, "loading model...", color=(255, 255, 255), scale=2)
lcd.display(img)
task = kpu.load(model_addr)
kpu.init_yolo2(task, 0.5, 0.3, 5, anchors) # threshold:[0,1], nms_value: [0, 1]
try:
while 1:
img = sensor.snapshot()
t = time.ticks_ms()
code = kpu.run_yolo2(task, img)
t = time.ticks_ms() - t
print(code)
if code:
from fpioa_manager import fm
from Maix import GPIO
io_water_beng = 11
fm.register(io_water_beng, fm.fpioa.GPIO0)
led_r=GPIO(GPIO.GPIO0, GPIO.OUT)
led_r.value(1)
for i in code:
a=img.draw_rectangle(i.rect(),(0,255,0),2)
a = lcd.display(img)
for i in code:
lcd.draw_string(i.x()+45, i.y()-5, labels[i.classid()]+" "+'%.2f'%i.value(), lcd.BLACK,lcd.GREEN)
else:
img.draw_string(0, 200, "t:%dms" %(t), scale=2, color=(255, 0, 0))
a = lcd.display(img)
except Exception as e:
raise e
finally:
kpu.deinit(task)
if name == "main":
try:
f=open("/sd/anchors.txt","r")\
anchor_txt=f.read()
L=[]
for i in anchor_txt.split(","):
L.append(float(i))
anchor=tuple(L)
f.close()
#main(anchors = anchor, model_addr=0x300000, lcd_rotation=2, sensor_window=(320, 240))
main(anchors = anchor, model_addr="/sd/yolov2.kmodel")
except Exception as e:
sys.print_exception(e)
lcd_show_except(e)
finally:
gc.collect()
达不到我想要的效果,不知道如何调代码。
参考下
https://blog.csdn.net/shikong2011/article/details/122810729