Image_Segmentation
Image_Segmentation copied to clipboard
No ConvTranspose layer in your code?
Hello, I found that there is ConvTrans layer in your network architecture. But i can not find ConvTranspose layer in the code you supported. Could you solve my confusion? Thank you!
class up_conv(nn.Module): def init(self,ch_in,ch_out): super(up_conv,self).init() self.up = nn.Sequential( nn.Upsample(scale_factor=2), nn.Conv2d(ch_in,ch_out,kernel_size=3,stride=1,padding=1,bias=True), nn.BatchNorm2d(ch_out), nn.ReLU(inplace=True) )
def forward(self,x):
x = self.up(x)
return x
Upsample