Import Error about "chamfer" Package

[Guaishou74851]

Hi, I downloaded your code on my 2 Ubuntu servers and google colab, but after I get the success of running these commands:

pip install -r requirements.txt
# Chamfer Distance
bash install.sh
# PointNet++
pip install "git+git://github.com/erikwijmans/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
# GPU kNN
pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl

Then I run your provided test example:

bash ./scripts/test.sh 0 \
    --ckpts ./pretrained/PoinTr_PCN.pth \
    --config ./cfgs/PCN_models/PoinTr.yaml \
    --exp_name example

but got:

bash ./scripts/test.sh 0 \
>     --ckpts ./pretrained/PoinTr_ShapeNet55.pth \
>     --config ./cfgs/ShapeNet55_models/PoinTr.yaml \
>     --mode easy \
>     --exp_name example
+ GPUS=0
+ PY_ARGS='--ckpts ./pretrained/PoinTr_ShapeNet55.pth --config ./cfgs/ShapeNet55_models/PoinTr.yaml --mode easy --exp_name example'
+ CUDA_VISIBLE_DEVICES=0
+ python main.py --test --ckpts ./pretrained/PoinTr_ShapeNet55.pth --config ./cfgs/ShapeNet55_models/PoinTr.yaml --mode easy --exp_name example

Traceback (most recent call last):
  File "main.py", line 1, in <module>
    from tools import run_net
  File "/userhome/chenbin/PoinTr/tools/__init__.py", line 1, in <module>
    from .runner import run_net
  File "/userhome/chenbin/PoinTr/tools/runner.py", line 5, in <module>
    from tools import builder
  File "/userhome/chenbin/PoinTr/tools/builder.py", line 8, in <module>
    from models import build_model_from_cfg
  File "/userhome/chenbin/PoinTr/models/__init__.py", line 2, in <module>
    import models.TopNet
  File "/userhome/chenbin/PoinTr/models/TopNet.py", line 6, in <module>
    from extensions.chamfer_dist import ChamferDistanceL2
  File "/userhome/chenbin/PoinTr/extensions/chamfer_dist/__init__.py", line 10, in <module>
    import chamfer
ModuleNotFoundError: No module named 'chamfer'

I tried for a long time to fix this problem, but still failed, so I have to ask you for help.
Could you please tell me how to fix it (or provide a simple colab example)? I just want to test and reproduce the results.
Looking forwarding to your response :) !

[yuxumin]

@Guaishou74851. Hi, the issue #6 is the same problem as you. So sorry that the bug wastes your time. I will mention this issue in readme to avoid wasting time on the implementation.