CanMV K210物体检测之串口通讯

import sensor, image, time, lcd, gc, cmath
from maix import KPU
from fpioa_manager import fm
from machine import UART

lcd.init()                          # Init lcd display
lcd.clear(lcd.RED)                  # Clear lcd screen.

# sensor.reset(dual_buff=True)      # improve fps
sensor.reset()                      # Reset and initialize the sensor.
sensor.set_pixformat(sensor.RGB565) # Set pixel format to RGB565 (or GRAYSCALE)
sensor.set_framesize(sensor.QVGA)   # Set frame size to QVGA (320x240)
sensor.set_vflip(0)              # 翻转摄像头
sensor.set_hmirror(0)            # 镜像摄像头
sensor.skip_frames(time = 1000)     # Wait for settings take effect.
clock = time.clock()                # Create a clock object to track the FPS.

fm.register(6, fm.fpioa.UART2_RX)
fm.register(8, fm.fpioa.UART2_TX)

yb_uart = UART(UART.UART2, 115200, 8, 0, 0, timeout=1000, read_buf_len=4096)

bytes1 = 'a'
bytes2 = 'b'

print("ready load model")

labels = ["good","bad"] #类名称，按照label.txt顺序填写
anchor = (2.38, 2.31, 3.25, 3.09, 3.94, 3.81, 4.75, 4.62, 5.64, 5.42) # anchors,使用anchor.txt中第二行的值

kpu = KPU()
# 从sd或flash加载模型
kpu.load_kmodel('/sd/det.kmodel')
#kpu.load_kmodel(0x300000, 584744)
kpu.init_yolo2(anchor, anchor_num=(int)(len(anchor)/2), img_w=320, img_h=240, net_w=320 , net_h=240 ,layer_w=10 ,layer_h=8, threshold=0.6, nms_value=0.3, classes=len(labels))

while(True):
    gc.collect()

    clock.tick()
    img = sensor.snapshot()

    kpu.run_with_output(img)
    dect = kpu.regionlayer_yolo2()

    fps = clock.fps()

    if len(dect) > 0:
        for l in dect :
            a = img.draw_rectangle(l[0],l[1],l[2],l[3],color=(0,255,0))

            info = "%s %.3f" % (labels[l[4]], l[5])
            #print(labels[l[4]])
            if(labels[l[4]]=="good"):
                yb_uart.write(bytes1)
                #print(labels[l[4]])
            elif(labels[l[4]]=="bad"):
                yb_uart.write(bytes2)
                #print(labels[l[4]])
            a = img.draw_string(l[0],l[1],info,color=(255,0,0),scale=2.0)
            del info

    a = img.draw_string(0, 0, "%2.1ffps" %(fps),color=(0,60,255),scale=2.0)
#    lcd.display(img)

这是根据嘉楠官网训练后生成的det.py脚本改的...注意存到SD卡中要重命名为main.py。

以上demo场景是二分类，识别到“good”返回a，识别到“bad”返回“b”。具体根据自己的需求修改，仅供参考。