wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-06-27 16:00:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated broken imports for new model modules

Browse Source
This commit is contained in:
wassname
2018-04-10 10:51:11 +08:00
parent 5fbc7effa6
commit 96f46514d6
3 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/__init__.py
+4 -5
View File
@@ -1,5 +1,4 @@
from .Pointnet2SemSeg import Pointnet2MSG as Pointnet2SemMSG
from .Pointnet2SemSeg import Pointnet2SSG as Pointnet2SemSSG
from .Pointnet2Cls import Pointnet2MSG as Pointnet2ClsMSG
from .Pointnet2Cls import Pointnet2SSG as Pointnet2ClsSSG
from .pointnet2_msg_sem import Pointnet2MSG as Pointnet2SemMSG
from .pointnet2_ssg_sem import Pointnet2SSG as Pointnet2SemSSG
from .pointnet2_msg_cls import Pointnet2MSG as Pointnet2ClsMSG
from .pointnet2_ssg_cls import Pointnet2SSG as Pointnet2ClsSSG
train_cls.py
+1 -1
View File
@@ -10,7 +10,7 @@ import os
import tensorboard_logger as tb_log
from models import Pointnet2ClsMSG as Pointnet
from models.Pointnet2Cls import model_fn_decorator
from models.pointnet2_msg_cls import model_fn_decorator
from data import ModelNet40Cls
import utils.pytorch_utils as pt_utils
import data.data_utils as d_utils
train_sem_seg.py
+2 -2
View File
@@ -9,8 +9,8 @@ import numpy as np
import tensorboard_logger as tb_log
import os
from models import Pointnet2SSG as Pointnet
from models.Pointnet2SemSeg import model_fn_decorator
from models import Pointnet2SemMSG as Pointnet
from models.pointnet2_msg_sem import model_fn_decorator
from data import Indoor3DSemSeg
import utils.pytorch_utils as pt_utils