项目需求:检测板子并画框。
遇到问题:物体未完全被抓手放平就圈框了。
思路:抓手未完全放下时检测有无抓手,有的话不画框。
解决方法:读取视频中的每一帧,截取想检测区域,根据蓝色的hsv范围提取成mask格式图像,遍历mask数组,若有非零元素说明提取成功,表示识别出了蓝色。用了800左右测试用例,几乎全部分类准确。
import cv2
import time
import numpy as np
cap = cv2.VideoCapture('C:/Users/10737/Desktop/GB/board.mp4')
def hsv_blue(img):
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
low_hsv = np.array([107,190,80])#可查找函数(点击鼠标获取某点的hsv或BGR值并打印,自己总结蓝色范围)
high_hsv = np.array([155,255,120])
mask = cv2.inRange