python 图片裁剪，图片主体颜色

最新推荐文章于 2024-03-26 08:58:28 发布

原创

最新推荐文章于 2024-03-26 08:58:28 发布 · 721 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一种基于Python的颜色识别算法，该算法可以识别图片中的主要颜色，并通过裁剪图片来进一步处理。涉及的技术包括使用OpenCV进行颜色范围定义、图片读取与转换、轮廓检测等。

from PIL import Image
import sys
import numpy as np
import collections
import  cv2
import numpy as np


#定义字典存放颜色分量上下限
#例如：{颜色: [min分量, max分量]}
#{'red': [array([160,  43,  46]), array([179, 255, 255])]}
 
def getColorList():
    dict = collections.defaultdict(list)
 
    # 黑色
    lower_black = np.array([0, 0, 0])
    upper_black = np.array([180, 255, 46])
    color_list = []
    color_list.append(lower_black)
    color_list.append(upper_black)
    dict['black'] = color_list
 
    # #灰色
    # lower_gray = np.array([0, 0, 46])
    # upper_gray = np.array([180, 43, 220])
    # color_list = []
    # color_list.append(lower_gray)
    # color_list.append(upper_gray)
    # dict['gray']=color_list
 
    # 白色
    lower_white = np.array([0, 0, 221])
    upper_white = np.array([180, 30, 255])
    color_list = []
    color_list.append(lower_white)
    color_list.append(upper_white)
    dict['white'] = color_list
 
    #红色
    lower_red = np.array([156, 43, 46])
    upper_red = np.array(