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()