python中批量将矩形图片转化为正方形图片，并且缩放

最新推荐文章于 2022-08-25 11:24:02 发布

Virogous

最新推荐文章于 2022-08-25 11:24:02 发布

阅读量2.7k

点赞数 7

分类专栏： pytorch 文章标签： python 深度学习

本文链接：https://blog.csdn.net/qq_40303258/article/details/106872056

版权

在使用PyTorch进行深度学习时，为确保输入是正方形图片，本文提供了一段Python代码，实现批量将矩形图片转换并缩放为正方形格式，适用于样本预处理。

摘要由CSDN通过智能技术生成

前言

为了在pytorch中进行深度学习的时候尽可能的采用正方形图片作为输入，需要将自己做的样本图片最好转化为正方形图片。具体的见代码，代码中有每一行的注释。

python代码

from PIL import Image
import os


def Square_Generated (read_file): # 创建一个函数用来产生所需要的正方形图片转化
    image = Image.open(read_file)   # 导入图片
    w, h = image.size  # 得到图片的大小
    # print(w,h)
    new_image = Image.new('RGB'