工创视觉——Openmv(开源)

打印 上一主题 下一主题

主题 888|帖子 888|积分 2664



目次
前言
一、结果展示
二、完整代码展示
三、关键词

前言

各位小伙伴大家好,今天我来分享一下我所利用的工创视觉代码。openmv做视觉是非常方便的,因为他有各种集成好的API,功能都已经封装好了,我们要利用什么功能就直接调用对应的API就可以了,并且只要我们了解API的参数的含义,就可以调解到我们想要的结果(openmv官网有对API参数的表明 )。openmv对于视觉小白们来说,照旧容易上手的。下面我就利用openmv来实现工创视觉任务。本人学识尚浅,如有不对的地方,还望各位大佬批评指正,谢谢。
阅读保举:
有一些底子的小伙伴们可以先大要看一遍代码,然后在继承任意看看本人的浅见。
没有底子或者没了解过openmv的小伙伴。可以直接利用我提供的代码去运行,整体代码我也调节测试过了。
一、结果展示

色环






二、完整代码展示

在代码中我已经标注了详细的解释,假如另有不理解的参数,可以去openmv的文档看一下。
  1. #视觉部分代码
  2. #整体以状态机方式运行
  3. import sensor, image
  4. import  time,json
  5. import struct
  6. from machine import UART
  7. from pyb import UART
  8. import pyb
  9. uart=UART(3,9600)     #初始化串口3
  10. #状态机初始状态
  11. flag=0x02
  12. ACK=0x00
  13. #默认为0x00无色,0x01为红色,0x02为蓝色,0x03为绿色
  14. colour=0x01
  15. #工作变量次数
  16. ACK_num=0
  17. #发送数据包
  18. def send_data(x,y):
  19.     #包头 数据位 状态机状态 目标级别 x坐标 y坐标 包尾
  20.     UART_Date=bytearray([
  21.                         0XFF,
  22.                         0x04,
  23.                         flag,
  24.                         colour,
  25.                         x,
  26.                         y,
  27.                         0XFE])
  28.     print(UART_Date)
  29.     uart.write(UART_TXDATA)
  30. #应答数据包
  31. def send_ACK():
  32.     #包头 数据位 应答位 包尾
  33.     UART_Date=bytearray([
  34.                         0XFF,
  35.                         0x01,
  36.                         0x11,
  37.                         0XFE])
  38.     print(UART_Date)
  39.     uart.write(UART_TXDATA)
  40. #接收数据包
  41. def RX_date():
  42.     if len(data) >= 5:
  43.         if data[0] == 0x00 and data[1] == 0x02 and data[4]==0x11:
  44.             data1 = data[2]
  45.             data2 = data[3]
  46.             print("数据1:", data1, "数据2:", data2)
  47.         else:
  48.             data1=0x00
  49.             data2=0x00
  50.             print("数据格式错误")
  51.     else:
  52.         data1=0x00
  53.         data2=0x00
  54.         print("接收到的数据长度不足")
  55. sensor.reset()
  56. sensor.set_pixformat(sensor.RGB565)
  57. sensor.set_framesize(sensor.QQVGA)
  58. sensor.set_auto_whitebal(False)
  59. sensor.skip_frames(30)
  60. sensor.set_auto_gain(False)
  61. #色环阈值#红、蓝、绿
  62. ring_colour=[(40, 71, 11, 40, -1, 69),
  63.                 (28, 62, -72, 20, -81, -22),
  64.                 (56, 77, -53, -18, -72, 33)]
  65. Binarzation=[(100,255)]#二值化阈值
  66. while(True):
  67.     #图像帧
  68.     img = sensor.snapshot()
  69.     #二维码查找部分
  70.     if flag==0x00:
  71.         #在二维码部分要关闭自动白平衡部分
  72.         #sensor.set_auto_whitebal(False)
  73.         #桶形畸变,调节下面这个函数参数大小即可
  74.         img.lens_corr(1.8)
  75.         #寻找二维码
  76.         for code in img.find_qrcodes():
  77.             #打印二维码信息
  78.             #code数组说明:0为x坐标,1为y坐标,2像素宽,3像素高,4二维码内容
  79.             print(code)
  80.             #串口发送部分(hex格式为准)
  81.             UART_TXDATA=bytearray([0XFF,code[0],code[1],code[2],code[3],0XFE])
  82.             uart.write(UART_TXDATA)
  83.             #等待下位机确认信息并发送ACK
  84.             if uart.any():
  85.                 ACK=uart.read()
  86.             #状态机置1
  87.             if ACK==0x11:
  88.                 ACK=0x00
  89.                 flag=1
  90.     #物流识别拿取部分
  91.     #后期还要调节最小像素大小threshold,根据摄像头高度调节
  92.     #货物颜色阈值
  93.     #(30, 57, 25, 54, 10, 42)红色机器阈值
  94.     #(0, 57, -128, 53, -128, -35)蓝色机器阈值
  95.     #(59, 87, -73, -31, -128, 62)绿色机器阈值
  96.     if flag==0x01:
  97.         #降低图像分辨率避免openmv物理空间栈不足
  98.         sensor.reset()
  99.         sensor.set_pixformat(sensor.RGB565)
  100.         sensor.set_framesize(sensor.QQVGA)
  101.         sensor.skip_frames(time = 2000)
  102.         #关闭摄像头自动增益
  103.         sensor.set_auto_gain(False)
  104.         #关闭自动白平衡
  105.         sensor.set_auto_whitebal(False)
  106.         clock = time.clock()
  107.         #一次拿取3次物品,应答3次就退出当前状态
  108.         while(True):
  109.             #图像帧
  110.             img = sensor.snapshot()
  111.             #红部分
  112.             if colour==0x01:
  113.                 for c in img.find_circles(threshold = 3500, x_margin = 10, y_margin = 10, r_margin = 10,
  114.                         r_min = 2, r_max = 100, r_step = 2):
  115.                     area = (c.x()-c.r(), c.y()-c.r(), 2*c.r(), 2*c.r())
  116.                     #area为识别到的圆的区域,即圆的外接矩形框
  117.                     #像素颜色统计
  118.                     statistics = img.get_statistics(roi=area)
  119.                     #打印呀圆形信息
  120.                     print(statistics)
  121.                     #阈值调节
  122.                     if 30<statistics.l_mode()<57 and 25<statistics.a_mode()<54 and 10<statistics.b_mode()<42:
  123.                         #白色的圆框出来
  124.                         img.draw_circle(c.x(), c.y(), c.r(), color = (255, 255, 255))
  125.                         #打印坐标信息
  126.                         print(c.x(),c.y())
  127.                 #打印运行帧率
  128.                 print("FPS %f" % clock.fps())
  129.             #蓝部分
  130.             if colour==0x02:
  131.                 for c in img.find_circles(threshold = 3500, x_margin = 10, y_margin = 10, r_margin = 10,
  132.                         r_min = 2, r_max = 100, r_step = 2):
  133.                     area = (c.x()-c.r(), c.y()-c.r(), 2*c.r(), 2*c.r())
  134.                     #area为识别到的圆的区域,即圆的外接矩形框
  135.                     #像素颜色统计
  136.                     statistics = img.get_statistics(roi=area)
  137.                     #打印呀圆形信息
  138.                     print(statistics)
  139.                     #阈值调节
  140.                     if 0<statistics.l_mode()<57 and -128<statistics.a_mode()<53 and -128<statistics.b_mode()<-35:
  141.                         #白色的圆框出来
  142.                         img.draw_circle(c.x(), c.y(), c.r(), color = (255, 255, 255))
  143.                         #打印坐标信息
  144.                         print(c.x(),c.y())
  145.                 #打印运行帧率
  146.                 print("FPS %f" % clock.fps())
  147.             #绿部分
  148.             if colour==0x03:
  149.                 for c in img.find_circles(threshold = 3500, x_margin = 10, y_margin = 10, r_margin = 10,
  150.                         r_min = 2, r_max = 100, r_step = 2):
  151.                     area = (c.x()-c.r(), c.y()-c.r(), 2*c.r(), 2*c.r())
  152.                     #area为识别到的圆的区域,即圆的外接矩形框
  153.                     #像素颜色统计
  154.                     statistics = img.get_statistics(roi=area)
  155.                     #打印呀圆形信息
  156.                     print(statistics)
  157.                     #阈值调节
  158.                     if 59<statistics.l_mode()<87 and -73<statistics.a_mode()<31 and -128<statistics.b_mode()<62:
  159.                         #白色的圆框出来
  160.                         img.draw_circle(c.x(), c.y(), c.r(), color = (255, 255, 255))
  161.                         #打印坐标信息
  162.                         print(c.x(),c.y())
  163.                 #打印运行帧率
  164.                 print("FPS %f" % clock.fps())
  165.             if ACK_num==3:
  166.                 ACK_num=0
  167.                 flag=2
  168.                 break
  169.     if flag==0x02:
  170.         #一共识别三次色环
  171.         while True:
  172.             #图像帧
  173.             img = sensor.snapshot()
  174.             #红部分
  175.             if colour==0x01:
  176.                 img.binary([ring_colour[0]])#红色阈值,图像二值化
  177.                 #滤波器的直径是2,0.1较小的值意味着将只模糊与中心像素颜色相近的像素,1滤波器在颜色空间中的敏感度
  178.                 img.bilateral(2, color_sigma=0.1, space_sigma=1)#双边滤波
  179.                 #卷积核越大运算量越大
  180.                 img.dilate(5)#卷积核为5x5的膨胀
  181.                 #graythreshold颜色阈值,pixels_threshold的最大像素数量,area_threashold的最小像素数量,merge允许相近图像重叠合并
  182.                 #返回一个符合图像对象的列表
  183.                 blobs=img.find_blobs(Binarzation,pixels_threshold=2025,area_threashold=1600,merge=True)
  184.                 #判断满足要求的对象是否只有一个
  185.                 if len(blobs)==1:
  186.                     #比较色块大小,取最大
  187.                     Blobmax=blobs[0]
  188.                     for blob in blobs:
  189.                         if blob.pixels() > Blobmax.pixels():
  190.                             Blobmax=blob
  191.                         else:
  192.                             Blobmax=Blobmax
  193.                     #返回边沿信息并画框
  194.                     img.draw_rectangle(Blobmax.rect())
  195.                     #打印中心坐标S
  196.                     print(Blobmax.cx(),Blobmax.cx())
  197.             #蓝部分
  198.             if colour==0x02:
  199.                 img.binary([ring_colour[1]])#蓝色阈值,图像二值化
  200.                 #滤波器的直径是2,0.1较小的值意味着将只模糊与中心像素颜色相近的像素,1滤波器在颜色空间中的敏感度
  201.                 img.bilateral(2, color_sigma=0.1, space_sigma=1)#双边滤波
  202.                 #卷积核越大运算量越大
  203.                 img.dilate(5)#卷积核为5x5的膨胀
  204.                 #graythreshold颜色阈值,pixels_threshold的最大像素数量,area_threashold的最小像素数量,merge允许相近图像重叠合并
  205.                 #返回一个符合图像对象的列表
  206.                 blobs=img.find_blobs(Binarzation,pixels_threshold=2025,area_threashold=1600,merge=True)
  207.                 #判断满足要求的对象是否只有一个
  208.                 if len(blobs)==1:
  209.                     #比较色块大小,取最大
  210.                     Blobmax=blobs[0]
  211.                     for blob in blobs:
  212.                         if blob.pixels() > Blobmax.pixels():
  213.                             Blobmax=blob
  214.                         else:
  215.                             Blobmax=Blobmax
  216.                     #返回边沿信息并画框
  217.                     img.draw_rectangle(Blobmax.rect())
  218.                     #打印中心坐标S
  219.                     print(Blobmax.cx(),Blobmax.cx())
  220.             #绿部分
  221.             if colour==0x03:
  222.                 img.binary([ring_colour[2]])#绿色阈值
  223.                 #滤波器的直径是2,0.1较小的值意味着将只模糊与中心像素颜色相近的像素,1滤波器在颜色空间中的敏感度
  224.                 img.bilateral(2, color_sigma=0.1, space_sigma=1)#双边滤波
  225.                 #卷积核越大运算量越大
  226.                 img.dilate(5)#卷积核为5x5的膨胀
  227.                 #graythreshold二值化阈值,pixels_threshold的最大像素数量,area_threashold的最小像素数量,merge允许相近图像重叠合并
  228.                 #返回一个符合图像对象的列表
  229.                 blobs=img.find_blobs(Binarzation,pixels_threshold=2025,area_threashold=1600,merge=True)
  230.                 #判断满足要求的对象是否只有一个
  231.                 if len(blobs)==1:
  232.                     #比较色块大小,取最大
  233.                     Blobmax=blobs[0]
  234.                     for blob in blobs:
  235.                         if blob.pixels() > Blobmax.pixels():
  236.                             Blobmax=blob
  237.                         else:
  238.                             Blobmax=Blobmax
  239.                     #返回边沿信息并画框
  240.                     img.draw_rectangle(Blobmax.rect())
  241.                     #打印中心坐标S
  242.                     print(Blobmax.cx(),Blobmax.cx())
复制代码
三、关键词

帧率

实在假如有利用openmv做过一些小玩具或者项目标小伙伴们应该都发现了一个小题目,那就是他的帧率。每次当我们编写步调复杂了一些,他的帧率就会特别低。帧率低代表的他的图像流也会很小,普通一点就是机器反应变慢了。那我们要是想进步机器反应的速度,那我们可以优化我们的代码,比如淘汰数据空间栈,减小图像处置惩罚的卷积核大小,减小运算量等等。都是是可以进步帧率的办法。
例如(减小图像处置惩罚卷积核大小):

在这个代码里面我利用的卷积核大小为5X5的,如许他的结果就会好一些,不过同时也会低落帧率的大小。有利有弊,假如你的openmv提示你的空间栈不足,你可以把卷积核大小改成3X3的,如许应该就可以了。
阈值

颜色阈值设置,我们可以利用openmv的阈值编辑器来调节出符合的结果。调节技巧,可以在我已经调节好的阈值上面举行微调,正负3的调节。
例如(每次调解阈值都要上电检测比对结果):
要调节红色物体的阈值
调节前参考标准:(30, 57, 25, 54, 10, 42)
调节过程:(33, 60, 28, 57, 13, 45)
发现结果变好,那就再次增加
调节过程:(36, 63, 31, 60, 16, 48)
结果开始变差
那就知道阈值结果差不多在(33-36, 60-63, 28-31, 57-60, 13-16, 45-48)之间。

色环

实在这个色环一开始我利用的处置惩罚办法和物资的处置惩罚办法一样都是利用霍尔圆API直接检测的,但是我发现结果不行(网络模型练习应该也是不可以的)。以是我跑到网上去拜读了几篇处置惩罚的办法,就有了现在简洁的办法。
色环检测思绪:
图像帧的颜色筛选并二值化——再举行简朴的滤波——再举行膨胀(连接每个小色环,使其为整体)——再利用API返回处置惩罚后的图像帧(包罗了辨认到的对象)——末了再利用API筛选出全部对象中最大的色块(这个色块就是色环)
物资

物资辨认就应该没啥可以说的了,就是把颜色阈值调节好和把要辨认的霍尔圆像素大小设置好,应该就没啥了。(可以装车了在调节一遍)

二维码

二维码我利用的就是openmv官网的历程,辨认结果看起来还不错。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

正序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

莱莱

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表