python中把图片从矩形处理成正方形

燕小硕

于 2020-05-04 15:08:34 发布

阅读量4.7k

点赞数 5

分类专栏：笔记文章标签：神经网络机器学习深度学习 pytorch 图像识别

本文链接：https://blog.csdn.net/weixin_44928646/article/details/105917645

版权

笔记专栏收录该内容

6 篇文章

订阅专栏

Python中把图片从矩形处理成正方形

在训练神经网络时，数据集中有的图片是矩形的，但是要求是正方形的，所以要对图片进行处理，如果直接缩放成正方形，会改变宽和高的比例，导致失真。
这里用的方法：把图片粘贴到一张正方形图片上，然后再缩放。
在这里插入图片描述

from PIL import Image
image=Image.open("1.jpg")
image = image.convert('RGB')
w, h = image.size
background = Image.new('RGB', size=(max(w, h), max(w, h)), color=(127, 127, 127))  # 创建背景图，颜色值为127
length = int(abs(w - h) // 2)  # 一侧需要填充的长度
box = (length, 0) if w < h else (0, length)  # 粘贴的位置
background.paste(image, box)
image_data=background.resize((256,256))#缩放
background.show()