本文主要是介绍遍历图片修改像素点值,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
我的目的: 将下图中除黑色掩码部分外的背景全部换成白色。
我的做法:遍历图片中像素点rgb三通道值之和不小于65的点,赋值(255,255,255)(白色)。关于为什么选择三通道像素值和小于65,通过画图软件中颜色选取器,掩码边缘的rgb值并不是纯粹的(0,0,0)黑色。
最后结果:
提示:选择cv2.imwirte保存图片不会改变图片尺寸。 plt.savefig保存会改变图片尺寸。
import os
import cv2
from PIL import Image
import numpy as np
img_root = "D:/study/test_data/black_mask"
save_root = "D:/study/test_data/only_mask"
fileList = os.listdir(img_root)
n = 0
# load image
for image in os.listdir(img_root):img_path = os.path.join(img_root, image)img = Image.open(img_path)img1 = np.array(img.convert('RGB'))sq = img1.shapelen = sq[0]width = sq[1]for i in range(0, len):for j in range(0, width):if not img1[i, j].sum() <= 65:img1[i, j] = (255, 255, 255)cv2.imwrite(os.path.join(save_root,fileList[n]), img1)n+=1
这篇关于遍历图片修改像素点值的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!