K210 MaixPy AI 功能 之 检测行人(带视频效果、数据集和模型)

2021-02-281.1万
AI 快速预览详细 收起
本文介绍了如何使用K210 MaixPy实现行人检测功能。提供了582张已标注的行人数据集,并展示了详细的代码示例和视频效果。通过K210芯片和MaixPy库,可以轻松实现对行人的实时检测。适合AI初学者和开发者,帮助快速上手并部署实际项目。



> K210 MaixPy AI 功能 之 检测 行人(带视频效果、数据集和模型)

作为开源组的划水摸鱼大佬鼠看到了 Sipeed 家的人出现了,我也配合发个贴吧。

我就不发简单的玩意了,我转发点有意思的内容过来。

源头:https://cn.bbs.sipeed.com/d/487

582张行人的数据集,224*224的,已经标注好了。大家拿去训练的话也希望把模型分享出来。





功能代码如下:(比较简单)

import sensor,image,lcd,time
import KPU as kpu
lcd.init(freq=15000000)
sensor.reset(dual_buff=1)
sensor.set_pixformat(sensor.RGB565)
sensor.set_framesize(sensor.QVGA)
sensor.set_hmirror(1)
sensor.set_vflip(0)
sensor.set_windowing((224, 224))

sensor.run(1)
clock = time.clock()
classes = ['person']
task = kpu.load(0x300000)
anchor = (0.76,1.98, 1.20,3.06, 1.51,4.30, 2.06,5.32, 2.87,6.02)
a = kpu.init_yolo2(task, 0.3, 0.3, 5, anchor)
while(True):
    clock.tick()
    img = sensor.snapshot()
    code = kpu.run_yolo2(task, img)
    print(clock.fps())
    fps =clock.fps()
    img.draw_string(2,2, ("%2.1ffps" %(fps)), color=(0,255,0), scale=2)
    if code:
        for i in code:
            a=img.draw_rectangle(i.rect())
            a = lcd.display(img)
            print(i.classid(),i.value())
            for i in code:
                lcd.draw_string(i.x(), i.y(), classes[i.classid()], lcd.RED, lcd.WHITE)
                lcd.draw_string(i.x(), i.y()+12, '%.2f'%i.value(), lcd.RED, lcd.WHITE)
    else:
        a = lcd.display(img)
a = kpu.deinit(task)


SunFebruary-202102288162..png (4.38 KB, 下载次数: 5840)

K210 MaixPy AI 功能 之 检测行人 - 视频效果与数据集_image_3.webp

SunFebruary-202102287415..png (26.04 KB, 下载次数: 5950)

K210 MaixPy AI 功能 之 检测行人 - 视频效果与数据集_image_6.webp

personmodel.zip

1.95 MB, 下载次数: 9965

模型训练数据集

person.zip

7.35 MB, 下载次数: 12291

模型与测试代码

创作许可协议

本项目采用 None(不开放任何权利,保留所有权利) 进行许可。

评论(7)
藏好小马脚
藏好小马脚
大佬,您的训练次数是多少嘞????
hottirger
hottirger
太厉害了
花生编程
花生编程
不错不错
花生编程
花生编程
厉害厉害
温暖的蓝色
温暖的蓝色
双击666!
gm-ai
gm-ai
双击666!
empty
empty
给技术部打call
Sipeed-大佬鼠
Sipeed-大佬鼠
作者
回复empty
感谢!
- 没有更多了 -