Python获取图片像素BGR值并生成纯色图
目录
- 前言
- 依赖安装
- 代码
- 验证一下
前言
最近工作有个需求,获取某张图片某个像素颜色,生成该颜色的纯色图片。所以写了一个工具,分享给大家,如果大家也有一样的场景,可以直接使用。
依赖安装
需要使用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[x, y] blue = img[x, y, 0] green = img[x, y, 1] red = img[x, y, 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、做了简单的文件后缀校验,如需修改,可以自己添加。
验证一下
准备的图片
执行结果
最终的图片
到此这篇关于Python获取图片像素BGR值并生成纯色图的文章就介绍到这了,更多相关Python生成纯色图内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们!
赞 (0)