Opencv 颜色过滤 Python mask 选取颜色颜色直方图

布满荆棘的人生 2022-09-16 04:59 180阅读 0赞

截图目标，分析目标的颜色直方图：

import cv2
    from matplotlib import pyplot as plt
    
    
    def show_img(path):
        img = cv2.imread(path)
        b, g, r = img[:, :, 0], img[:, :, 1], img[:, :, 2]
        hist_b = cv2.calcHist([b], [0], None, [256], [0, 256])
        hist_g = cv2.calcHist([g], [0], None, [256], [0, 256])
        hist_r = cv2.calcHist([r], [0], None, [256], [0, 256])
        plt.figure(figsize=(10, 5))
        plt.subplot(121)
        plt.plot(hist_r, color='r', label="r")
        plt.plot(hist_g, color='g', label="g")
        plt.plot(hist_b, color='b', label="b")
        plt.legend()
        img2 = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
        h, s, v = img2[:, :, 0], img2[:, :, 1], img2[:, :, 2]
        hist_h = cv2.calcHist([h], [0], None, [256], [0, 256])
        hist_s = cv2.calcHist([s], [0], None, [256], [0, 256])
        hist_v = cv2.calcHist([v], [0], None, [256], [0, 256])
        plt.subplot(122)
        plt.plot(hist_h, color='r', label="h")
        plt.plot(hist_s, color='g', label="s")
        plt.plot(hist_v, color='b', label="v")
        plt.legend()
    
        plt.show()
        return hist_r, hist_g, hist_b, hist_h, hist_s, hist_v
    
    
    r, g, b, h, s, v = show_img("lanse.png")

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAUTc0Mjk3MTYzNg_size_20_color_FFFFFF_t_70_g_se_x_16]

import cv2
    import numpy as np
    from matplotlib import pyplot as plt
    
    img = cv2.imread("1111.jpg")
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    
    lower_red = np.array([115, 0, 160])
    upper_red = np.array([140, 30, 190])
    mask = cv2.inRange(hsv, lower_red, upper_red)
    
    
    res = cv2.bitwise_and(img, img, mask=mask)
    # cv2.imshow('img', img)
    cv2.imshow('mask', mask)
    cv2.imshow('res', res)
    
    cv2.waitKey(0)
    cv2.destroyAllWindows()

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBAUTc0Mjk3MTYzNg_size_20_color_FFFFFF_t_70_g_se_x_16]: /images/20220828/f903be00964e4117b3513e701d09a249.png