网络编程
位置：首页>> 网络编程>> Python编程>> python opencv根据颜色进行目标检测的方法示例

python opencv根据颜色进行目标检测的方法示例

作者：Abc2149　　发布时间：2021-09-29 03:53:41　

标签：python,opencv,目标检测

颜色目标检测就是根据物体的颜色快速进行目标定位。使用cv2.inRange函数设定合适的阈值，即可以选出合适的目标。

建立项目colordetect.py，代码如下：


#! /usr/bin/env python
# -*- coding: utf-8 -*-

import numpy as np
import cv2

def colorDetect():
image = cv2.imread('./1.png')
# 使用RGB颜色空间检测红 蓝 黄 灰，设置合适的阈值
boundaries = [
([17, 15, 100], [50, 56, 200]),
([86, 31, 4], [220, 88, 50]),
([25, 146, 190], [62, 174, 250]),
([103, 86, 65], [145, 133, 128])
]

for lower, upper in boundaries:
lower = np.array(lower, dtype='uint8')
upper = np.array(upper, dtype='uint8')
# 低于lower和高于upper的像素为黑色，lower-upper之间的像素为白色
mask = cv2.inRange(image, lower, upper)
# 利用蒙版，进行图像的逻辑与运算
output = cv2.bitwise_and(image, image, mask=mask)

cv2.imshow('image', np.hstack([image, output]))
cv2.waitKey(0)
cv2.destroyAllWindows()

def main():
colorDetect()

if __name__ == "__main__":
main()

定义RGB颜色列表：


boundaries = [
([17, 15, 100], [50, 56, 200]),
([86, 31, 4], [220, 88, 50]),
([25, 146, 190], [62, 174, 250]),
([103, 86, 65], [145, 133, 128])
]

该部分([17, 15, 100], [50, 56, 200])，表示图像像素R>=100, B>=15, G>=15和R<=200, B<=56, G<=50的像素将视为红色。

执行代码，结果如下：

python opencv根据颜色进行目标检测的方法示例

总结

要检测图像中颜色，第一件事要做的就是定义像素值的上限和下限。不同的颜色空间具有不同上下限值，定义了上限和下限后，就可以调用cv2.inRange方法返回一个mask，将该mask与图像进行逻辑与bitwise_and就可以得到该图像。

参考资料
https://www.pyimagesearch.com

来源：https://blog.csdn.net/qq_36281080/article/details/103926608

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com