Skip to content


Browse files Browse the repository at this point in the history
  • 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/
Expand Up @@ -9,23 +9,11 @@ def cls_network(octree, flags, training, reuse=False):
return network_ocnn(octree, flags, training, reuse)
elif == 'resnet':
return network_resnet(octree, flags, training, reuse)
elif == 'hrnet':
return HRNet(flags).network_cls(octree, training, reuse)
elif == 'cnn_grids':
return network_cnn_grids(octree, flags, training, reuse)
elif == 'resnet_grids':
return network_resnet_grids(octree, flags, training, reuse)
print('Error, no network: ' +

def seg_network(octree, flags, training, reuse=False, pts=None, mask=None):
if == 'unet':
return network_unet(octree, flags, training, reuse)
elif == 'hrnet':
return HRNet(flags).network_seg(octree, training, reuse, pts, mask)
# elif == 'unet_scannet':
# return network_unet34(octree, flags, training, reuse, pts, mask)
print('Error, no network: ' +

0 comments on commit 2ae9f59

Please sign in to comment.