Problems with object part segmentation

[l53ma]

Thanks for sharing your great work!

I am a beginner of deep learning models used on point clouds. I am very interested about your achievement. I can successfully conducted classification operation using your codes. However, I have one question when conducting object part segmentation operation on ShapeNetPart dataset.

After running "cd part_seg", and "python train.py", it output an Error message as following:

Traceback (most recent call last):
File "train.py", line 47, in
MODEL = importlib.import_module(FLAGS.model) # import network module
File "/usr/lib/python2.7/importlib/init.py", line 37, in import_module
import(name)
ImportError: No module named model

My operating environment is Ubuntu16.04 + python2.7, can you help me address this issue?

Thank you for your kind assistance.

[mf27]

It's a bug in the code. In the file train.py, change the line:
parser.add_argument('--model', default='model', help='Model name [default: model]')
to
parser.add_argument('--model', default='pointnet2_part_seg', help='Model name [default: model]')