最近要做一个深度学习项目,但是发现数据集大小不太规整,或者是不是我想要的尺寸,于是决定批量修改统一大小。
【原来的数据】
原来的数据大小是20×20的,我想要的是32×32大小的。
【修改尺寸后的数据】
可以看到尺寸明显变大了。
脚本如下:
import cv2
import os
# 原始图片文件夹路径
ori_path = 'D:\\pycharm\\certify_code_detect\\datasets\\car_data\\0\\'
# 修改后保存的文件夹路径
save_path = 'D:\\pycharm\\certify_code_detect\\datasets\\new_data\\0\\'
num = os.listdir(ori_path) # num为list类型,存储着每张图片的名字
print('共{}张图片'.format(len(num))) # 计算机当前目录有几张图片
for i in range(len(num)):
img = cv2.imread(ori_path+num[i]) # 读取
print(i)
resize = cv2.resize(img, (32, 32)) # resize
cv2.imwrite(save_path+str(i)+'.jpg',resize) # 保存
cv2.waitKey(0)
脚本不复杂,但是实用。