python批量处理矩形转为正方形并完成缩放
发布时间
阅读量：
阅读量
前言
在使用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', size=(max(w, h), max(w, h)),color= 'white') # 创建新的一个图片，大小取长款中最长的一边，color决定了图片中填充的颜色
# print(background)
length = int(abs(w - h)) # 一侧需要填充的长度
box = (length, 0) if w < h else (0, length) # 放在box中
new_image.paste(image, box) 全部评论 (0)
还没有任何评论哟~