python获取图片的rgb值 python提取图片颜色

依赖安装
需要使用opencv以及numpy。安装命令如下：

pip install opencv-python -i https://pypi.douban.com/simple
pip install numpy -i https://pypi.douban.com/simple
代码
#!/user/bin/env python

coding=utf-8

“”"
@project : csdn
@author : 剑客阿良_ALiang
@file : make_pic_tool.py
@ide : PyCharm
@time : 2022-01-11 08:34:31
“”"
import cv2
import os
import numpy as np
import uuid

获取图片坐标bgr值

def get_pix_bgr(image_path: str, x: int, y: int):
ext = os.path.basename(image_path).strip().split(‘.’)[-1]
if ext not in [‘png’, ‘jpg’]:
raise Exception(‘format error’)
img = cv2.imread(image_path)
px = img[y, x]
blue = img[y, x, 0]
green = img[y, x, 1]
red = img[y, x, 2]
return blue, green, red

构建纯色图

def make_one_color_pic(output_dir: str, image_path: str, coordinates: tuple, resolution: tuple):
blue, green, red = get_pix_bgr(image_path, coordinates[0], coordinates[1])
img = np.zeros((resolution[1], resolution[0], 3), np.uint8)
# 创建BGR纯色图
img[:] = [blue, green, red]
result_image = os.path.join(output_dir, ‘{}.jpg’.format(uuid.uuid1().hex))
cv2.imwrite(result_image, img)
return result_image

if name == ‘main’:
print(make_one_color_pic(r’C:\Users\huyi\Desktop’, r’C:\Users\huyi\Desktop\2054146.jpg’, (300, 300), (1080, 1920)))
代码说明：

1、get_pix_bgr方法入参分别为，图片地址以及坐标位置，用以获取bgr值。

2、make_one_color_pic方法为最终生成纯色图方法，参数有输出目录地址、图片地址、坐标位置、最终图片分辨率，输出最终图片路径。

3、最终图片名使用uuid，避免重复。

4、做了简单的文件后缀校验，如需修改，可以自己添加。