怎么提取图片里的颜色?图像颜色特征提取

图像的颜色一直是研究和关注的热点,也是特征工程不可或缺的feature,今天就简单介绍一种非常基础的颜色的rgb特征。

  1. 目的:

我们希望图片的rgb特征转化成一个可衡量的指标,例如人的身高、体重 这样的指标173cm, 60kg,用于后续工作。那么怎么转化呢?首先我们需要定一个标准,这里是从图片中提取了8种颜色(红、橙、黄....)用来反映图片的RGB指标, 就好比表述一个人的健康指标,可能回身高、体重、血常规等等。

2.颜色提取:

首先我们可以在网上找到hex颜色和对应名称的网页,如下:

通过html解析,我们就可以获得hex颜色和颜色名称的对应关系,然后再通过hex -> rgb转化就可以得到 rgb和颜色名称,如下表:

 

 ### 颜色转化方法
    def hex_to_int(s):
    	""" hex to rgb"""
        s = s.lstrip("#")
        return int(s[:2], 16), int(s[2:4], 16), int(s[4:6], 16)

 

有了对应关系后,我们需要抽提出我们关心的颜色,如红色,黄色等,但问题就是红色可能会有很多种,这时就需要我们队将相近颜色进行合并。

def distance(coord1, coord2):
    """ 相近颜色计算方法 """
    return math.sqrt(sum([(i - j)**2 for i, j in zip(coord1, coord2)]))

我们先来测试一下相近颜色计算方法:

(1)我们比较一下距离: 红色 vs 绿色 距离 是不是 大于 红色 vs 粉色 ? 结果发现红色和粉色更接近,说明我们可以按照这样的方法进行。

 

distance(colors['red'], colors['green']) # 273.70787347096905
distance(colors['red'], colors['pink']) # 232.76812496559748
print(f"{distance(colors['red'], colors['green']) > distance(colors['red'], colors['pink'])}") ## True

(2)比较一下颜色:就是把每个颜色相近的颜色拿出来,大致检查一下即可,方法如下:

#### 通过相似度算法进行相近颜色合并
all_imgs = []
for color_name in ['red', 'orange', 'yellow', 'green', 'blue', 'purple', 'grey', 'pink']:
    topk = 5
    imgs = []
    closest_names = closest(colors, colors[color_name], topk)
    closest_colors = list(map(colors.get, closest_names))
    
    print(f"{color_name}: {closest_names}")
    for i, (name, rgb) in enumerate(zip(closest_names, closest_colors)):
        img = show_color(rgb, size=100)
        imgs.append(pil2cv_img(img))
    imgs = np.hstack(imgs)
    all_imgs.append(imgs)
all_imgs = np.vstack(all_imgs)

cv2_pilimg(all_imgs)

 

这样我们就拿到了指标颜色。有了指标颜色,后面的就简单了。我们先那一张图片测试一下颜色提取效果:

 

这样的很简单的就可以拿到rgb特征了。当然,方法有很多种,这只是其中的一种。

今天就分享到这里, 我是paperClub。

备注:WeiXin 搜索paperClub, 添加关注并回复【paperClub】即可获取使用方法。 

1. 感谢各位小伙伴的关注, 您的点赞、鼓励和留言,都是我深夜坚持的一份动力,无论褒贬,都是我们行进途中最好的回馈,也都会被认真对待。

2. 我将持续分享各类、好玩且有趣的算法应用及工程和项目,欢迎分享和转发。沟通、学习和交流,请与我联系,虽然平时忙,但留言必回,勿急,感谢理解!

3. 分享内容包括开源项目和自研项目,如在引用或使用时,考虑不周、遗漏引用信息或涉及版权等,请您及时联系。如果您对某些内容感兴趣,我们可以一起讨论、交流和学习。
 

  • 1
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值