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)