【Pytorch】自定义的transfomer模块(torchvision)---ZeroPaddingResize

た入场券 2022-10-16 08:43 22阅读 0赞

# 1.目的 #

有时候直接进行resize会有形变，所以想到这样的方式，同比例缩放，然后补0。

# 2.实现 #

参考：https://www.pythonf.cn/read/144858

class ZeroPaddingResize(object):
        """
            resize image with same scale, fill with zero padding
        """
        def __init__(self, size=(224, 224), interpolation=Image.BILINEAR):
            self.size = size
            self.interpolation = interpolation
    
        def __call__(self, img):
            ih, iw = img.size
            h, w = self.size[0], self.size[1]
            scale = min(w / iw, h / ih)
            new_w = int(iw * scale+0.5)
            new_h = int(ih * scale+0.5)
    
            img = img.resize((new_w, new_h), self.interpolation)
            new_img = Image.new('RGB', self.size, (0, 0, 0))
            new_img.paste(img, ((w-new_w)//2, (h-new_h)//2))
    
            return new_img

# 3.使用 #

import torchvision
    import ZeroPaddingResize
    
    transform_train = torchvision.transforms.Compose([
        ZeroPaddingResize((224, 224)),
        torchvision.transforms.ToTensor(),
    ])

发表评论取消回复

表情：

评论列表（有 0 条评论，22人围观）

还没有评论，来说两句吧...

相关阅读

相关自定义模块

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ub

ゞ浴缸里的玫瑰/ 2023年10月05日 12:21/ 0 赞/ 104 阅读

相关 Pytorch自定义数据集

简述 Pytorch自定义数据集方法，应该是用pytorch做算法的最基本的东西。往往网络上给的demo都是基于torch自带的MNIST的相关类。所以，为了解决使

我就是我/ 2023年07月24日 02:55/ 0 赞/ 260 阅读

相关 php自定义模块,ThinkPHP5 安装自定义模块

安装官方给的demo，在build.php文件中 // +----------------------------------------------------------

Love The Way You Lie/ 2023年01月15日 08:25/ 0 赞/ 247 阅读

相关自定义模块_多彩的花——自定义模块

程序设计中通过旋转已有的图片，然后重复执行，画笔模块库中这种应用非常多。这个程序设计是应用图章的命令模块，然后通过重复执行和设置X坐标的设置，这样就可以很容易画出多彩的话。

小灰灰/ 2023年01月06日 14:57/ 0 赞/ 300 阅读

相关【Pytorch】自定义的transfomer模块(torchvision)---ZeroPaddingResize

1.目的有时候直接进行resize会有形变，所以想到这样的方式，同比例缩放，然后补0。 2.实现参考：https://www.pythonf.cn/read/

た入场券/ 2022年10月16日 08:43/ 0 赞/ 23 阅读

相关自定义模块

自定义模块 Python中有许多模块是系统自带的，也可以自己编写模块，自定义模块。所谓的自定义模块，就是将写好的python程序(后缀名为.py)放在lib

女爷i/ 2022年06月07日 13:47/ 0 赞/ 319 阅读

相关自定义模块

需要创建一个node\_modules文件夹，然后再里面定义模块，例如：m6模块，在下面定义js和package.json ![20180103092115565][]

雨点打透心脏的1/2处/ 2022年06月02日 04:10/ 0 赞/ 352 阅读

相关 Pytorch 自定义数据集

在Pytorch中定义数据集主要涉及到两个主要的类：Dataset、DataLoader。 Dataset类 `Dataset`类是`Pytorch`中所有数据集加载类

川长思鸟来/ 2022年01月21日 06:35/ 0 赞/ 477 阅读

相关自定义模块

1. 模块的定义与分类模块是什么？这几天，我们进入模块的学习。在学习模块之前，我们首先要知道，什么是模块？一个函数封装一个功能，你使用的软件可能就是由n多

蔚落/ 2021年12月23日 12:01/ 0 赞/ 518 阅读

相关 pytorch 自定义网络结构 +自定义数据加载

自定义网络结构 import os os.environ['CUDA_VISIBLE_DEVICES'] = '3' os.system('e

àì夳堔傛蜴生んèń/ 2021年10月31日 16:26/ 0 赞/ 505 阅读