search

charlesq34 / pointnet2

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space
Other
3.01k stars 889 forks source link

Problems with object part segmentation #40

Open l53ma opened 6 years ago

l53ma commented 6 years ago

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 commented 6 years ago

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]')