def __init__(self): super(DecoderHead, self).__init__() self.output_size = self._hyper_params["output_size"] self.out_projector = projector(128, 1) self.f_s16_projector = projector(256, 1) self.f_s8_projector = projector(256, 1) self.activation = nn.Sigmoid()
def __init__(self): super(DecoderHead, self).__init__() self.output_size = self._hyper_params["output_size"] self.upblock1 = upsample_block(512, 512, 256) self.upblock2 = upsample_block(256, 256, 256) self.upblock3 = upsample_block(256, 128, 256) self.upblock4 = upsample_block(256, 64, 128) self.out_projector = projector(128, 1) self.f_s16_projector = projector(256, 1) self.f_s8_projector = projector(256, 1) self.out_projector = projector(128, 1) self.activation = nn.Sigmoid()
def __init__(self, basemodel): super(JointEncoder, self).__init__() self.basemodel = basemodel self.projector_corr_feature = projector(256, 256)