Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Update network_factory.py
  • Loading branch information
Qinqing Liu committed Dec 20, 2021
1 parent a6d4f6d commit 2ae9f59
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions tensorflow/script/network_factory.py
Expand Up @@ -9,23 +9,11 @@ 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':
return network_resnet_grids(octree, flags, training, reuse)
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)


0 comments on commit 2ae9f59

Please sign in to comment.