问题:我要做一些夜间图像数据集,但是找到的图像的分辨率都太大了,都是19201080的,需要对数据预处理一下,我按照比例缩放之后分辨率应该为:480270。防止我写完的代码以后用到我找不到,将python代码贴到博客:
from PIL import Image
import glob
from torchvision import transforms
import torchvision
#修改目录中的图像分辨率
data_trans=transforms.Compose([
torchvision.transforms.ToTensor()
])
width=480 #设置想要的宽度
height=270 #设置想要的高度
namelist=glob.glob(r'E:\Data\Dark_Zurich_train_anon\rgb_anon\train\night\GP010376\*.png') # 获取文件夹下的文件列表
i=2089 #这里是文件一共2089,我懒得再设置个变量了,直接初始化了2089,初始计数
for name in namelist:
img = Image.open(name)
out = img.resize((width, height), Image.ANTIALIAS) # resize image with high-quality
out.save(f'E:/Data/Dark_Zurich_train_anon/rgb_anon/train/night/test/{i}.jpg') #保存
print(f'{i}.jpg 保存成功')
img2 = data_trans(out)
print(img2.shape)
i=i+1