filaPro / oneformer3d

[CVPR2024] OneFormer3D: One Transformer for Unified Point Cloud Segmentation
Other
348 stars 32 forks source link

Import error while using Docker #10

Closed sonukiller closed 10 months ago

sonukiller commented 10 months ago

When running the repo using docker, I am getting, ImportError: Failed to import custom modules from {'imports': ['oneformer3d']}. I also checked mmengine custom import from here, but couldn't be resolved. Can you please help with this?

filaPro commented 10 months ago

Can you please share the command you are running and the full log? I think adding PYTHONPATH=./ before python test.py ... should help.

sonukiller commented 10 months ago

Thanks, it is working!

russellyq commented 8 months ago

Can you please share the command you are running and the full log? I think adding PYTHONPATH=./ before python test.py ... should help.

Hi @filaPro ,

After adding "PYTHONPATH=./ python ....." when running the command, I still cannot run the code.

FYI, I set up my conda env locally with mmdet3d

oneformer3d-contributor commented 8 months ago

Can you please share the full log including error message?

Also what do you mean by with mmdet3d, is it just pip install ... mmdet3d ...?

russellyq commented 8 months ago

Can you please share the full log including error message?

Also what do you mean by with mmdet3d, is it just pip install ... mmdet3d ...?

Hi @oneformer3d-contributor ,

Thanks for your quick action.

I solved it myself. It is related to my python env settings.

By the way, when training on S3DIS, I find that the cuda memory varies from time to time. Even when running with batch size 1, GPU runs out of memory. I run on RTX4090.

Any answer to this ?

Thanks

oneformer3d-contributor commented 8 months ago

There are some very large scenes on S3DIS and we don't use crops. You can try with 32Gb GPU or follow #11.

russellyq commented 8 months ago

There are some very large scenes on S3DIS and we don't use crops. You can try with 32Gb GPU or follow #11.

Hi @oneformer3d-contributor,

Thank you. I have successfully ran the code on S3DIS after resolving all the related issues.

May I ask for any codes on how to only evaluate on validation set ?

oneformer3d-contributor commented 8 months ago

python tools/test.py ... commands in the readme are basically for the evaluation on validation set.