This commit is contained in:
Peng-Shuai Wang 2021-03-23 16:16:26 +08:00 коммит произвёл GitHub
Родитель 58da15dc80
Коммит 594226ee65
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

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