bubbliiiing/unet-pytorch

关于upsample的方法

peterchen183 opened this issue · 0 comments

import torch
import torch.nn as nn

from nets.resnet import resnet50
from nets.vgg import VGG16


class unetUp(nn.Module):
    def __init__(self, in_size, out_size):
        super(unetUp, self).__init__()
        self.conv1  = nn.Conv2d(in_size, out_size, kernel_size = 3, padding = 1)
        self.conv2  = nn.Conv2d(out_size, out_size, kernel_size = 3, padding = 1)
        self.up     = nn.UpsamplingBilinear2d(scale_factor = 2)
        ...

您好作者,这里选择的是nn.UpsamplingBilinear2d,是否有可能替换成nn.ConvTranspose2d?
能否指点一二,感谢!