python+opencv识别四种颜色，框定并标注

最新推荐文章于 2024-08-05 23:08:48 发布

呔吗

最新推荐文章于 2024-08-05 23:08:48 发布

阅读量9.1k

点赞数 7

文章标签： opencv python 计算机视觉

本文链接：https://blog.youkuaiyun.com/weixin_60874964/article/details/123436748

版权

import cv2 as cv
import numpy as np

# 读取视频
capture = cv.VideoCapture(0)
height = capture.get(cv.CAP_PROP_FRAME_HEIGHT)
width = capture.get(cv.CAP_PROP_FRAME_WIDTH)
count = capture.get(cv.CAP_PROP_FRAME_COUNT)
fps = capture.get(cv.CAP_PROP_FPS)
print(height, width, count, fps)

ball_color = ['green','red','blue']

def process(image, opt=1):
    # RGB转HSV色彩空间
    hsv = cv.cvtColor(image, cv.COLOR_BGR2HSV)
    # 结构元素
    line = cv.getStructuringElement(cv.MORPH_RECT, (15, 15), (-1, -1))
    # HSV范围
    mask_red = cv.inRange(hsv, (0,60,60), (50,255,255))
    mask_blue = cv.inRange(hsv, (100, 80, 46), (124,255,255))
    mask_green = cv.inRange(hsv, (35,43,35), (90,255,255))
    mask_yellow = cv.inRange(hsv,(11,34,43),(46,255,255))

    # 开操作
    masks = [mask_red,mask_blue,mask_green,mask_yellow]
    
    # 轮廓提取, 发现最大轮廓
    i = 1
    for mask in masks:

        contours, hierarchy = cv.findCo