Exemple #1
0
    def get_segmentation_net(self):

        encoder = self.encoder
        segmentation_decoder = self.segmentation_decoder

        encoder.return_gradient_reversal = False
        segmentation_decoder.return_gradient_reversal = False

        return unet_from_encoder_decoder(encoder, segmentation_decoder)
Exemple #2
0
    def get_segmentation_net(self):

        return unet_from_encoder_decoder(self.encoder,
                                         self.segmentation_decoder)