更改图片通道、重命名图片、修改图片像素值、更改图片尺寸

 修改图片通道

import os
import cv2
import tqdm
import numpy as np
from PIL import Image


gt_path = './gtFine_trainvaltest/gtFine/train/qdu/'


gtlist = os.listdir(gt_path)
for gt in tqdm.tqdm(gtlist):
    read_path = os.path.join(gt_path, gt)
    
    mask = Image.open(read_path)
    img = np.array(mask)
    print(img.shape, mask.size)

    #导入 opencvimport cv2  #读入原始图像，使用 cv2.IMREAD_UNCHANGED
    img = cv2.imread(read_path, cv2.IMREAD_UNCHANGED)  #查看原始图像
    shape = img.shape
    if len(shape) == 2:
     continue
    # print(shape)

    if shape[2] == 3 or shape[2] == 4:      
     img_gray = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)    #将彩色图转化为单通道图
    # print(img_gray.shape) #查看转换后图像通道
    # cv2.imshow("gray_img", img_gray)
    # cv2.imshow("image", img)
    # cv2.waitKey(0)
    # cv2.destroyAllWindows()

    cv2.imwrite(read_path, img_gray)

    # break

批量命名

import os
import tqdm
from PIL import Image


gt_path = './gtFine_trainvaltest/gtFine/train/qdu/'
img_path = './leftImg8bit_trainvaltest/leftImg8bit/train/qdu/'

gt_suffix = '_gtFine_labelIds.png'
img_suffix = '_leftImg8bit.png'

gtlist = os.listdir(gt_path)
for gt in tqdm.tqdm(gtlist):

    newname = gt[:-4] + gt_suffix
    src = os.path.join(gt_path, gt)
    dst = os.path.join(gt_path, newname)
    os.rename(src, dst)

    newname = gt[:-4] + img_suffix
    src = os.path.join(img_path, gt)
    dst = os.path.join(img_path, newname)
    os.rename(src, dst)

缩放尺寸

import os
import tqdm
from PIL import Image


gt_path = './gtFine_trainvaltest/gtFine/train/qdu/'
img_path = './leftImg8bit_trainvaltest/leftImg8bit/train/qdu/'

gtlist = os.listdir(gt_path)
for gt in tqdm.tqdm(gtlist):

    rdpath = os.path.join(gt_path, gt)
    mask = Image.open(rdpath)
    mask = mask.resize((2048, 1024), Image.ANTIALIAS)
    # Image.NEAREST ：低质量
    # Image.BILINEAR：双线性
    # Image.BICUBIC ：三次样条插值
    # Image.ANTIALIAS：高质量
    mask.save(rdpath)

    gt = gt.replace('_gtFine_labelIds.png', '_leftImg8bit.png')
    rdpath = os.path.join(img_path, gt)
    image = Image.open(rdpath)
    image = image.resize((1024, 2048), Image.ANTIALIAS)
    image.save(rdpath)

修改像素值

import os
import cv2
import tqdm
import glob
import numpy as np
from PIL import Image

np.set_printoptions(threshold=np.inf)
gt_path = './gtFine_trainvaltest/gtFine/train/qdu/'
gtlist = os.listdir(gt_path)

for gt in tqdm.tqdm(gtlist):

    read_path = os.path.join(gt_path, gt)
    mask = Image.open(read_path)


    img = np.array(mask)
    img[img < 100] = 0
    img[img > 100] = 1

    image = Image.fromarray(img)
    image.save(read_path)

#    img = np.array(mask)
#    print(img.shape, img)
 
    break

修改亮度

import os
from tqdm import tqdm
import PIL.Image as Image
input_path = './leftImg8bit_trainvaltest/leftImg8bit/train_origine/qdu/'
dim_ouput_path = './leftImg8bit_trainvaltest/leftImg8bit/train_dark/qdu/'
bright_ouput_path = './leftImg8bit_trainvaltest/leftImg8bit/train_bright/qdu/'


img_suffix = '_leftImg8bit.png'
dark_img_suffix = '_dark_leftImg8bit.png'
bright_img_suffix = '_bright_leftImg8bit.png'

for id, name in tqdm(enumerate(os.listdir(input_path))):

    img_path = input_path+name
    img = Image.open(img_path)

    img_dim = img.point(lambda p: p*0.5)
    img_bright = img.point(lambda p: p*1.5)

    darkname = name.replace(img_suffix, dark_img_suffix)
    img_dim.save(dim_ouput_path + darkname)
    brightname = name.replace(img_suffix, bright_img_suffix)
    img_bright.save(bright_ouput_path + brightname)