transforms.Resize 和 transforms.CenterCrop的组合

最新推荐文章于 2024-09-18 17:42:18 发布

FrankJingle

最新推荐文章于 2024-09-18 17:42:18 发布

阅读量954

点赞数 3

分类专栏： Computer Vision Deep Learning 文章标签：深度学习 pytorch

本文链接：https://blog.csdn.net/fangjin_kl/article/details/128097071

版权

46 篇文章 1 订阅

订阅专栏

23 篇文章 1 订阅

订阅专栏

看到很多代码会有类似于这种数据增强变换

T = transforms.Compose([
            transforms.ToTensor(),
            transforms.Resize(120),
            transforms.CenterCrop(120),
        ])

有些疑惑，既然resize成120了，又crop一样大小，不是白忙活了吗？后来阅读了一下源码，发现，resize是取了一个最小的边给缩放到120，整体是不会变形的，并不是变成120120。再结合crop，才给变成120120,

具体见官方文档解释.

Parameters:
size (sequence or int) –
Desired output size. If size is a sequence like (h, w), output size will be matched to this. If size is an int, smaller edge of the image will be matched to this number. i.e, if height > width, then image will be rescaled to (size * height / width, size).

效果参考 https://www.jianshu.com/p/43b541caba7f