K210 MaixPy实现红绿灯检测

疫情在家参加了十五届智能车竞赛百度创意组的线上赛,参与了红绿灯检测,最近在玩K210,想着在K210上实现红绿灯检测。现将模型开源

数据集准备

百度AI Studio官方红绿灯检测比赛数据集点这里

我采用的是Sipeed的官方在线训练平台maixhub,选择了百度官方数据集其中1500张图片,这里就需要数据集修改成要求格式,具体要求及说明在这里

效果

https://www.bilibili.com/video/BV1yZ4y1G7dX

模型开源

链接:https://pan.baidu.com/s/1AxC1E5jCvVUD663X-JN9eQ
提取码:afhz

  • 2
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
在k210 maixpy上同时运行电动车检测模型和口罩检测模型,需要进行以下步骤: 1. 在k210 maixpy上安装电动车检测模型和口罩检测模型的程序和模型文件。 2. 在程序中导入电动车检测模型和口罩检测模型的模型文件,并将它们分别加载到k210芯片的内存中。 3. 编写程序,将k210芯片的摄像头捕获的图像数据输入到两个模型中进行判断,判断结果可以输出到显示屏或者其他设备上。 4. 在程序中设置两个模型的运行频率和优先级,确保它们可以同时运行且不会相互干扰。 5. 对程序进行测试和优化,确保它能够稳定运行,并能够准确地识别出电动车和口罩。 下面是一个简单的示例程序,演示了如何在k210 maixpy上同时运行电动车检测模型和口罩检测模型: ``` import sensor import image import lcd import KPU as kpu # 初始化摄像头和LCD显示屏 sensor.reset() sensor.set_pixformat(sensor.RGB565) sensor.set_framesize(sensor.QVGA) sensor.set_vflip(True) sensor.run(1) lcd.init() # 加载电动车检测模型和口罩检测模型 car_detect_model = kpu.load("/sd/models/car_detect.kmodel") mask_detect_model = kpu.load("/sd/models/mask_detect.kmodel") # 设置模型的运行频率和优先级 kpu.set_outputs(car_detect_model, 0, 1, 2) # 设置电动车检测模型的输出 kpu.set_outputs(mask_detect_model, 3, 4, 5) # 设置口罩检测模型的输出 kpu.set_freq(car_detect_model, 500000000) # 设置电动车检测模型的运行频率 kpu.set_freq(mask_detect_model, 300000000) # 设置口罩检测模型的运行频率 kpu.set_priority(car_detect_model, 1) # 设置电动车检测模型的优先级 kpu.set_priority(mask_detect_model, 2) # 设置口罩检测模型的优先级 while True: # 捕获摄像头的图像数据 img = sensor.snapshot() # 在电动车检测模型中进行判断 car_detect_results = car_detect_model.forward(img) # 在口罩检测模型中进行判断 mask_detect_results = mask_detect_model.forward(img) # 在LCD显示屏上显示结果 lcd.display(img) # 输出结果到串口或其他设备上 print("car detect results:", car_detect_results) print("mask detect results:", mask_detect_results) # 释放模型的内存 kpu.deinit(car_detect_model) kpu.deinit(mask_detect_model) ``` 在上面的示例程序中,我们首先初始化了摄像头和LCD显示屏,然后加载了电动车检测模型和口罩检测模型,并设置了它们的运行频率和优先级。接着,在程序的主循环中,我们捕获了摄像头的图像数据,并分别将它们输入到两个模型中进行判断。最后,我们将结果输出到LCD显示屏和串口上。注意,在程序运行结束后,我们需要释放模型的内存,避免内存泄漏。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值