Python獲取圖片像素BGR值並生成純色圖

Posted on by

前言

最近工作有個需求，獲取某張圖片某個像素顏色，生成該顏色的純色圖片。所以寫瞭一個工具，分享給大傢，如果大傢也有一樣的場景，可以直接使用。

依賴安裝

需要使用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生成純色圖內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：