使用opencv选择任意照片识别七彩颜色（python）

最新推荐文章于 2022-03-05 14:54:12 发布

kevin33824

最新推荐文章于 2022-03-05 14:54:12 发布

阅读量305

点赞数 1

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

本文链接：https://blog.csdn.net/weixin_43512867/article/details/121301700

版权

这个Python程序创建了一个简单的GUI，用户可以选择七种颜色（红色、蓝色、绿色、橙色、紫色、青色和黄色）之一，然后打开图像文件。程序会自动检测并标记出图像中对应颜色的区域。它使用了OpenCV和Matplotlib库进行颜色空间转换、颜色范围定义和图像显示。

摘要由CSDN通过智能技术生成

# 识别七彩颜色
import cv2 as cv
import matplotlib.pyplot as plt
import tkinter as tk
from tkinter import filedialog
import numpy as np
from tkinter import *


class Application(Frame):
    """一个经典的GUI程序的写法"""

    def __init__(self, master=None):
        super().__init__(master)  # super代表的是父类的定义，而不是父类对象
        self.master = master
        self.lower_hsv = np.array([0, 0, 0])
        self.upper_hsv = np.array([0, 0, 0])
        self.pack()
        self.createWidget()

    def createWidget(self):
        self.button1 = Button(self, text='红色', width=5, command=self.blue, anchor='c').pack(side='right')
        self.button2 = Button(self, text='蓝色', width=5, command=self.red, anchor='c').pack(side='left')
        self.button3 = Button(self, text='绿色', width=5, command=self.green, anchor='c').pack(side='left')
        self.button4 = Button(self, text='橙色', width=5, command=self.orange, anchor='c').pack(side='right')
        self.button5 = Button(self, text='紫色', width=5, command=self.purple, anchor='c').pack(side='left')
        self.button6 = Button(self, text='青色', width=5, command=self.cyan, anchor='c').pack(side='left')
        self.button7 = Button(self, text='黄色', width=5, command=self.yellow, anchor='c').pack(side='right')
        self.pack()

    def red(self):
        self.lower_hsv = np.array([0, 43, 46])
        self.upper_hsv = np.array([10, 255, 255])
        self.open()

    def blue(self):
        self.lower_hsv = np.array([100, 43, 46])
        self.upper_hsv = np.array([124, 255, 255])
        self.open()

    def green(self):
        self.lower_hsv = np.array([35, 43, 46])
        self.upper_hsv = np.array([77, 255, 255])
        self.open()

    def orange(self):
        self.lower_hsv = np.array([11, 43, 46])
        self.upper_hsv = np.array([25, 255, 255])
        self.open()

    def purple(self):
        self.lower_hsv = np.array([125, 43, 46])
        self.upper_hsv = np.array([155, 255, 255])
        self.open()

    def cyan(self):
        self.lower_hsv = np.array([78, 43, 46])
        self.upper_hsv = np.array([99, 255, 255])
        self.open()

    def yellow(self):
        self.lower_hsv = np.array([26, 43, 46])
        self.upper_hsv = np.array([34, 255, 255])
        self.open()

    @staticmethod
    def open():
        file_path = filedialog.askopenfilename()
        img = cv.imread(file_path)
        img2 = cv.cvtColor(img, cv.COLOR_BGR2HSV)  # 转换BGR色彩空间到HSV色彩空间
        mask = cv.inRange(img2, app.lower_hsv, app.upper_hsv)  # cv.inRange()函数的作用是提取想要的颜色，并把该颜色的区域设置为白色，其余的设置为黑色
        cnts1, hierarchy1 = cv.findContours(mask, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_NONE)  # 轮廓检测

        for cnt in cnts1:
            (x, y, w, h) = cv.boundingRect(cnt)  # 该函数返回矩阵四个点
            cv.rectangle(img, (x, y - 20), (x + w, y + h), (0, 0, 255), 2)  # 将检测到的颜色框起来

        plt.subplot(121), plt.imshow(img)
        plt.title('Image'), plt.xticks([]), plt.yticks([])
        plt.subplot(122), plt.imshow(mask)
        plt.title('Find Image'), plt.xticks([]), plt.yticks([])
        plt.show()


if __name__ == "__main__":
    app = Application()
    app.mainloop()