diff --git a/tensorflow/script/network_factory.py b/tensorflow/script/network_factory.py index 74e1164..23979a9 100644 --- a/tensorflow/script/network_factory.py +++ b/tensorflow/script/network_factory.py @@ -9,8 +9,6 @@ def cls_network(octree, flags, training, reuse=False): return network_ocnn(octree, flags, training, reuse) elif flags.name.lower() == 'resnet': return network_resnet(octree, flags, training, reuse) - elif flags.name.lower() == 'hrnet': - return HRNet(flags).network_cls(octree, training, reuse) elif flags.name.lower() == 'cnn_grids': return network_cnn_grids(octree, flags, training, reuse) elif flags.name.lower() == 'resnet_grids': @@ -18,14 +16,4 @@ def cls_network(octree, flags, training, reuse=False): else: print('Error, no network: ' + flags.name) -def seg_network(octree, flags, training, reuse=False, pts=None, mask=None): - if flags.name.lower() == 'unet': - return network_unet(octree, flags, training, reuse) - elif flags.name.lower() == 'hrnet': - return HRNet(flags).network_seg(octree, training, reuse, pts, mask) - # elif flags.name.lower() == 'unet_scannet': - # return network_unet34(octree, flags, training, reuse, pts, mask) - else: - print('Error, no network: ' + flags.name) -