遍历图片修改像素点值

我的目的： 将下图中除黑色掩码部分外的背景全部换成白色。

 我的做法：遍历图片中像素点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

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！