Python unet_from_encoder_decoder примеры использования

Пример #1

    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)

Пример #2

Файл: ynet_bvae.py Проект: torms3/domain-adaptive-segmentation

    def get_segmentation_net(self):

        return unet_from_encoder_decoder(self.encoder,
                                         self.segmentation_decoder)