遍历图片修改像素点值

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

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