зеркало из https://github.com/microsoft/O-CNN.git
Update network_factory.py
This commit is contained in:
Родитель
58da15dc80
Коммит
594226ee65
|
@ -3,7 +3,7 @@ from network_cls import network_ocnn, network_resnet
|
|||
from network_unet import network_unet
|
||||
from network_segnet import network_segnet
|
||||
from network_hrnet import HRNet
|
||||
from network_unet_scannet import network_unet34
|
||||
|
||||
|
||||
def cls_network(octree, flags, training, reuse=False):
|
||||
if flags.name.lower() == 'ocnn':
|
||||
|
@ -27,4 +27,4 @@ def seg_network(octree, flags, training, reuse=False, pts=None, mask=None):
|
|||
else:
|
||||
print('Error, no network: ' + flags.name)
|
||||
|
||||
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче