图像分割扣出掩码部分图像，背景换白色

最新推荐文章于 2023-12-26 01:15:00 发布

AKA_ZLJ

最新推荐文章于 2023-12-26 01:15:00 发布

阅读量664

点赞数 1

分类专栏：脚本代码文章标签： python 计算机视觉 numpy

本文链接：https://blog.csdn.net/weixin_63356640/article/details/129408041

版权

脚本代码专栏收录该内容

2 篇文章 0 订阅

订阅专栏

接上一篇博客：https://blog.csdn.net/weixin_63356640/article/details/129407332?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22129407332%22%2C%22source%22%3A%22weixin_63356640%22%7D

目的： 扣出图片中掩码部分图像，背景换成白色。

代码：

import os
import cv2
from PIL import Image
import numpy as np
img_root = "D:/data/grab/video_frames/frames_ce/3000" #图片路径
img_root1 ="D:/study/test_data/only_mask" #掩码路径
save_root = "D:/study/test_data/only_mask_png" #保存扣出掩码部分图片的路径
fileList = os.listdir(img_root)  #将文件夹中文件转为列表形式
fileList1= os.listdir(img_root1)
n = 0
# load image
for n in range(3000): #批量处理3000张图片
        image = Image.open(os.path.join(img_root, fileList[n])) #拼接图片路径，打开图片
        image1= Image.open(os.path.join(img_root1, fileList1[n]))
        img = np.array(image.convert('RGB')) #将图片转化为张量形式
        img1 = np.array(image1.convert('RGB'))
        sq = img1.shape #图片尺寸
        len = sq[0]
        width = sq[1]
        for i in range(0, len):
             for j in range(0, width):
                  if  img1[i, j].sum() <= 65:
                        img1[i, j] = img[i, j] #遍历图片，像素值小于65的部分赋值真实图像原始值
        img1 =cv2.cvtColor(img1, cv2.COLOR_BGR2RGB) #cv2读取的bgr通道转为rbg,否则图片偏蓝
        cv2.imwrite(os.path.join(save_root, fileList[n]), img1)
        n+=1