修改图片尺寸脚本(用于制作数据集使用)
看了一个19年的文章在他的基础上做了一定的修改,更符合现在的机器学习过程中的使用
原文链接
首先要有对应的python环境 python3版本以上请安装pillow
对应的指令为
pip install Pillow
然后建立相应的文件夹保存图片
这里我的文件路径为
folder_path ="C:\D\数据集"#图片的绝对路径
然后编写脚本
import os
from PIL import Image
folder_path ="C:\D\数据集"#图片的绝对路径,同理也别把这个放在图片文件夹里面
filt_list = os.listdir(folder_path)
def ResizeImage(filein,fileout,width,height,type):
img = Image.open(filein)
out = img.resize((width,height),Image.ANTIALIAS)
out.save(fileout,type)
if __name__ == "__main__":
os.chdir(folder_path)
for i in range(1,len(filt_list)+1):
filein = filt_list[i-1]
fileout = filt_list[i-1]
width = 640
height = 640
type = 'png'#没啥用,不用改但是又不能没有
ResizeImage(filein,fileout,width,height,type)
这里请尽量不要将该脚本放到与图片所在的目录