def _upsample(self, original_feature, scaled_feature, scale_factor=2):
# is this correct? You do lose information on the upscale...
height, width = scaled_feature.size()[2:]
return F.upsample(original_feature, scale_factor=scale_factor)[:, :, :height, :width]
评论列表
文章目录