前言
最近工作有个需求,获取某张图片某个像素颜色,生成该颜色的纯色图片。所以写了一个工具,分享给大家,如果大家也有一样的场景,可以直接使用。
依赖安装
需要使用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、做了简单的文件后缀校验,如需修改,可以自己添加。
验证一下
准备的图片
执行结果
最终的图片
总结
最近工作还是比较忙的,有空的话再多写写。
分享:
荣誉就像河流:轻浮的和空虚的荣誉浮在河面上,沉重的和厚实的荣誉沉在河底里。 ——《随笔集》
如果本文对你有帮助的话,点个赞吧,谢谢!
本人CSDN主页地址:剑客阿良_ALiang的主页
一起学习,一起进步。
来源:https://www.cnblogs.com/jk-aliang/p/15787637.html
图文来源于网络,如有侵权请联系删除。