为了账号安全,请及时绑定邮箱和手机立即绑定

有 python 识别颜色方块数量的第三方库推荐吗?

/ 猿问

有 python 识别颜色方块数量的第三方库推荐吗?

桃花长相依 2019-05-19 14:46:46

下图由黄,红,绿,蓝组成,如何获取每种颜色方块的数量?
如果用 python 来实现的话,有相关的库或例子吗,谢谢

https://img4.mukewang.com/5ce0fc450001bc9c08000334.jpg

查看完整描述

1 回答

?
哔哔one

首先,用Pillow把图片读到变量pic里面

from PIL import Image

img = Image.open("example.png")

pic = img.load()

然后, 得到图片的高度和宽度


w, h = img.width, img.height

然后,做个循环统计各种颜色数


result = {'y': 0, 'r': 0, 'g': 0, 'b': 0}


for i in range(w): 

    for j in range(h): 

        color = pic[i,j] 

        c = '?' 

        if color[0] > 127 and color[1] > 127: 

            c = 'y' 

        elif color[0] > 127: 

            c = 'r' 

        elif color[1] > 127: 

            c = 'g' 

        else: 

            c = 'b' 

        result[c] = result[c] + 1

我的结果是:


print(result)

> {'y': 50803, 'r': 10113, 'g': 45177, 'b': 161107}


# 总数也对的上

sum(result.values()) == w * h

> True


查看完整回答
反对 回复 2019-05-19

添加回答

回复

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信