K210人脸检测

╰+哭是因爲堅強的太久メ 2022-12-24 03:59 194阅读 0赞

人脸检测

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,0xff,0x00))
        lcd.display(img)
    
    #mnist.kmodel  "/sd/mnist.kmodel"  model_addr=0x300000
    #
    def main(model_addr="/sd/facedetect.kmodel", lcd_rotation=0, sensor_hmirror=False, sensor_vflip=False):
        sensor.reset()
        sensor.set_pixformat(sensor.RGB565)
        sensor.set_framesize(sensor.QVGA)
        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)
    
        #task = kpu.load(model_addr)0x300000"/sd/mnist.kmodel"
        task = kpu.load("/sd/facedetect.kmodel")
        #"/sd/mnist.kmodel"
        anchors = (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, anchors) # threshold:[0,1], nms_value: [0, 1]
        try:
            while(True):
                img = sensor.snapshot()
                img.rotation_corr(x_rotation=180,y_rotation=180)#x_rotation=10，
                t = time.ticks_ms()
                objects = kpu.run_yolo2(task, img)
                t = time.ticks_ms() - t
                if objects:
                    for obj in objects:
                        img.draw_rectangle(obj.rect())
                img.draw_string(0, 200, "t:%dms" %(t), scale=2)
                lcd.display(img)
        except Exception as e:
            raise e
        finally:
            kpu.deinit(task)
    
    
    if __name__ == "__main__":
        try:
            main( model_addr="/sd/facedetect.kmodel", lcd_rotation=0, sensor_hmirror=False, sensor_vflip=False)
            # main(model_addr="/sd/m.kmodel")
        except Exception as e:
            sys.print_exception(e)
            lcd_show_except(e)
        finally:
            gc.collect()

模型文件

可以将模型直接烧录到内部0x300000地址下也可以放到内存卡中

facedetect.kmode

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 1][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 2][]

具体的烧录模型  安装MaixPy IDE  和固件下载固件  具体参考 [https://cn.maixpy.sipeed.com/dev/zh][https_cn.maixpy.sipeed.com_dev_zh]

代码放在

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 3][]

固件

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 4][]

模型

只选择一个    带0x300000为下载到内部flash 中的

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 5][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70]: /images/20221120/058e5fe903b94524a9c5b1e09569b998.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 1]: /images/20221120/784f8683a65e46488c91781b3fe8c9f1.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 2]: /images/20221120/b1c02af32a12406bb0ff6b80c61a19e4.png
[https_cn.maixpy.sipeed.com_dev_zh]: https://cn.maixpy.sipeed.com/dev/zh/get_started/get_started_fs.html
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 3]: https://img-blog.csdnimg.cn/20201126180015670.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 4]: https://img-blog.csdnimg.cn/20201126180121482.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0_size_16_color_FFFFFF_t_70 5]: https://img-blog.csdnimg.cn/20201126180205858.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2OTU4MTA0,size_16,color_FFFFFF,t_70