采用Costas信号时频图
取源文件前100张图片
每张图片像素不一,无法进行训练
创建目标存放文件夹,记录两个文件夹的绝对路径,并带入python代码中
源文件夹绝对路径:'C:\\Users\\Lenovo\\Desktop\\lpi_signals\\Costas\\'
需要保存的文件夹绝对路径:'C:\\Users\\Lenovo\\Desktop\\Costas_preprocessing\\'
from torchvision import transforms
from PIL import Image
from torchvision.utils import save_image
path_resource = 'C:\\Users\\Lenovo\\Desktop\\lpi_signals\\Costas\\'
path_destination = 'C:\\Users\\Lenovo\\Desktop\\Costas_preprocessing\\'
for i in range(1, 101):
img = Image.open(path_resource+'/Costas'+str(i)+'.jpg', mode='r')
img_to_tensor = transforms.ToTensor()(img)
# 双线性三次插值
bicubic_interpolation = transforms.Resize(
size=(227, 227),
interpolation=transforms.InterpolationMode.BICUBIC)
bicubic_resize_img = bicubic_interpolation(img_to_tensor)
save_image(
bicubic_resize_img, path_destination+'bicubic_Costas'+str(i)+'.jpg')
# ./代表当前所在目录
# yuhan 2023.4.25
检验结果,前一百张Costas时频图已经缩小为统一的227*227大小,可以进行下一步的训练。