search

JiehongLin / SAM-6D

[CVPR2024] Code for "SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation".
258 stars 22 forks source link

ModuleNotFoundError: No module named 'pointnet2._ext' #19

Closed kevinDrawn closed 4 months ago

kevinDrawn commented 4 months ago

My Cuda Version is 11.5

When I ran the 'sh demo.sh', ModuleNotFoundError: No module named 'pointnet2._ext' appeared.

INFO:root:Moving models to cuda done! INFO:root:Initializing template /SAM-6D/SAM-6D/Instance_Segmentation_Model/run_inference_custom.py:83: DeprecationWarning: Starting with ImageIO v3 the behavior of this function will switch to that of iio.v3.imread. To keep the current behavior (and make this warning disappear) use import imageio.v2 as imageio or call imageio.v2.imread directly. depth = np.array(imageio.imread(depth_path)).astype(np.int32) set CUDA_VISIBLE_DEVICES as 0 => creating model ... Traceback (most recent call last): File "/SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/pointnet2_utils.py", line 26, in import pointnet2._ext as _ext ModuleNotFoundError: No module named 'pointnet2._ext' `

Can you give me any suggestion?

kevinDrawn commented 4 months ago

I solved it. I missed the setup.py in the pointnet2