Python 获取图片某像素BGR值并生成纯色图 | Python工具

Posted 2022-02-05 剑客阿良_ALiang

目录

前言

依赖安装

代码

总结

---

前言

最近工作有个需求，获取某张图片某个像素颜色，生成该颜色的纯色图片。所以写了一个工具，分享给大家，如果大家也有一样的场景，可以直接使用。

依赖安装

需要使用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、做了简单的文件后缀校验，如需修改，可以自己添加。

验证一下

准备的图片

 

执行结果

 

最终的图片

 

总结

最近工作还是比较忙的，有空的话再多写写。

分享：

       荣誉就像河流：轻浮的和空虚的荣誉浮在河面上，沉重的和厚实的荣誉沉在河底里。

                                                                                                                        ——《随笔集》

如果本文对你有帮助的话，点个赞吧，谢谢！