K210 MAIXPY AI功能：行人检测

  • 时间:
  • 分类: Python

582个数据集，224*224
视频网址：
Https://www.bilibili.com/video/BV1Ra4y1s7Ei
代码如下：

选择 | 换行 | 行号
  1. import sensor,image,lcd,time
  2. import KPU as kpu
  3. lcd.init(freq=15000000)
  4. sensor.reset(dual_buff=1)
  5. sensor.set_pixformat(sensor.RGB565)
  6. sensor.set_framesize(sensor.QVGA)
  7. sensor.set_hmirror(1)
  8. sensor.set_vflip(0)
  9. sensor.set_windowing((224, 224))
  10.  
  11.  
  12. sensor.run(1)
  13. clock = time.clock()
  14. classes = ['person']
  15. task = kpu.load(0x300000)
  16. anchor = (0.76,1.98, 1.20,3.06, 1.51,4.30, 2.06,5.32, 2.87,6.02)
  17. a = kpu.init_yolo2(task, 0.3, 0.3, 5, anchor)
  18. while(True):
  19.     clock.tick()
  20.     img = sensor.snapshot()
  21.     code = kpu.run_yolo2(task, img)
  22.     print(clock.fps())
  23.     fps =clock.fps()
  24.     img.draw_string(2,2, ("%2.1ffps" %(fps)), color=(0,255,0), scale=2)
  25.     if code:
  26.         for i in code:
  27.             a=img.draw_rectangle(i.rect())
  28.             a = lcd.display(img)
  29.             print(i.classid(),i.value())
  30.             for i in code:
  31.                 lcd.draw_string(i.x(), i.y(), classes[i.classid()], lcd.RED, lcd.WHITE)
  32.                 lcd.draw_string(i.x(), i.y()+12, '%.2f'%i.value(), lcd.RED, lcd.WHITE)
  33.     else:
  34.         a = lcd.display(img)
  35. a = kpu.deinit(task)

标签: python

评论已关闭