关于upsample的方法
peterchen183 opened this issue · 0 comments
peterchen183 commented
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?
能否指点一二,感谢!