python- 数据集批量处理图像大小、转灰度图、四通道转三通道

# -*- coding: utf-8 -*-
import os
import cv2
from PIL import Image
import numpy as npdef getFileList(dir, Filelist, ext=None):"""获取文件夹及其子文件夹中文件列表输入 dir：文件夹根目录输入 ext: 扩展名返回： 文件路径列表"""newDir = dirif os.path.isfile(dir):if ext is None:Filelist.append(dir)else:if ext in dir[-3:]:Filelist.append(dir)elif os.path.isdir(dir):for s in os.listdir(dir):newDir = os.path.join(dir, s)getFileList(newDir, Filelist, ext)return Filelistorg_img_folder = '数据集地址'# 检索文件
imglist = getFileList(org_img_folder, [], 'png')
print('本次执行检索到 ' + str(len(imglist)) + ' 张图像\n')for imgpath in imglist:print(imgpath)image = Image.open(imgpath)# image = image.resize((32, 32))   # 批量处理图像大小image = image.convert("RGB")  # 4通道转化为rgb三通道image.save(imgpath)# image = cv2.imread(imgpath, cv2.IMREAD_GRAYSCALE) #转单通道# cv2.imwrite(imgpath, image)image = np.array(image)print('image.shape', image.shape)

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