图片数据清洗中超级常用的一招操作！

点击上方“AI算法与图像处理”，选择加"星标"或“置顶”

重磅干货，第一时间送达

导读

 

在做图像分类的任务中，经常需要将图片resize到指定的尺寸，例如 224，直接resize的结果，会导致图片失真，因此需要对其进行填充操作。

例如我们需要对上面的图片进行resize，直接使用resize 操作得到的结果：

大小：224*224

显然并不是我们想要的结果。

我们希望得到的是：

1、图片大小缩放

2、图片特征不失真（保持长宽比）

保存长宽比的秘诀在于，在较短的一边填充黑色

from PIL import Image


def make_square(im, , fill_color=(0, 0, 0, 0)):
    x, y = im.size
    size = max(x, y)
    new_im = Image.new('RGB', (size, size), fill_color)
    new_im.paste(im, (int((size - x) / 2), int((size - y) / 2)))
    new_img = new_img.resize((224, 224))
    return new_im
    
test_image = Image.open('test.jpg')
new_image = make_square(test_image)
new_image.show()
new_image.save('result.jpg')

最终的结果：

下载1：何恺明顶会分享
在「AI算法与图像处理」公众号后台回复：何恺明，即可下载。总共有6份PDF，涉及 ResNet、Mask RCNN等经典工作的总结分析
下载2：leetcode 开源书
在「AI算法与图像处理」公众号后台回复：leetcode，即可下载。每题都 runtime beats 100% 的开源好书，你值得拥有！

下载3 CVPR2020


在「AI算法与图像处理」公众号后台回复：CVPR2020，即可下载1467篇CVPR 2020论文
个人微信（如果没有备注不拉群！）
请注明：地区+学校/企业+研究方向+昵称



觉得不错就点亮在看吧