K210 MaixPy AI 功能 之 检测行人（带视频效果、数据集和模型）
本帖最后由 Sipeed-大佬鼠 于 2021-2-28 14:36 编辑> K210 MaixPy AI 功能 之 检测 行人（带视频效果、数据集和模型）
作为开源组的划水摸鱼大佬鼠看到了 Sipeed 家的人出现了，我也配合发个贴吧。
我就不发简单的玩意了，我转发点有意思的内容过来。
源头：https://cn.bbs.sipeed.com/d/487
582张行人的数据集，224*224的，已经标注好了。大家拿去训练的话也希望把模型分享出来。
视频效果：https://www.bilibili.com/video/BV1Ra4y1s7Ei
https://www.bilibili.com/video/BV1Ra4y1s7Ei
功能代码如下：（比较简单）
```python
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, 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)
```
给技术部打call empty 发表于 2021-2-28 20:41
给技术部打call
感谢! 双击666！ 双击666！ 厉害厉害 不错不错 太厉害了
大佬，您的训练次数是多少嘞？？？？
页:
[1]