def __init__(self, config): super(SegmentationNet10aIsola, self).__init__() self.patch_side = config.isola_patch_side self.input_sz = config.input_sz self.features_sz = SegmentationNet10a.cfg[-1][0] print(f"SegmentationNet10aIsola: {self.patch_side}, {self.input_sz}, {self.features_sz}") self.features = SegmentationNet10aTrunk(config, cfg=SegmentationNet10a.cfg) self.isola_head = IsolaHead(config) self._initialize_weights()
def __init__(self, config): super(SegmentationNet10aTwoHead, self).__init__() self.batchnorm_track = config.batchnorm_track self.trunk = SegmentationNet10aTrunk(config, cfg=SegmentationNet10a.cfg) self.head_A = SegmentationNet10aHead(config, output_k=config.output_k_A, cfg=SegmentationNet10a.cfg) self.head_B = SegmentationNet10aHead(config, output_k=config.output_k_B, cfg=SegmentationNet10a.cfg) self._initialize_weights()
def __init__(self, config): super(SegmentationNet10aDoersch, self).__init__() self.patch_side = config.doersch_patch_side self.input_sz = config.input_sz self.features_sz = SegmentationNet10a.cfg[-1][0] print("SegmentationNet10aDoersch: %d %d %d" % (self.patch_side, self.input_sz, self.features_sz)) self.features = SegmentationNet10aTrunk(config, cfg=SegmentationNet10a.cfg) self.doersch_head = DoerschHead(config) self._initialize_weights()