(self)
| 369 | class autoencoder_vgg7(nn.Module): # no decoder |
| 370 | ''' vgg encoder with bilinear upsampling''' |
| 371 | def __init__(self): |
| 372 | super(autoencoder_vgg7, self).__init__() |
| 373 | self.encoder = models.vgg19(pretrained=True).features |
| 374 | |
| 375 | def forward(self, x, upsampleH=224, upsampleW=224): |
| 376 | feat_out = [] # we only use high level features |