ModuleNotFoundError: No module named 'lib.nms'

Question

ModuleNotFoundError: No module named 'lib.nms'

jinyl777 opened this issue 4 years ago · 10 comments

I follow your read me but sh lib/build.sh not work and I got ModuleNotFoundError: No module named 'lib.nms'

Answer 1 · 2020-08-26T07:40:23.000Z

What is the log output after you run the command sh lib/build.sh ?

Answer 2 · 2020-08-26T07:45:48.000Z

The output is :
g++ -pthread -shared -L/home/guow/anaconda3/envs/CTtrack/lib -Wl,-rpath=/home/guow/anaconda3/envs/CTtrack/lib,--no-as-needed -L/home/guow/anaconda3/envs/CTtrac/lib -Wl,-rpath=/home/guow/anaconda3/envs/CTtrack/lib,--no-as-needed build/temp.linux-x86_64-3.6/nms_kernel.o build/temp.linux-x86_64-3.6/gpu_nms.o -L/usr/loca/cuda/lib64 -L/home/guow/anaconda3/envs/CTtrack/lib -Wl,--enable-new-dtags,-R/usr/local/cuda/lib64 -lcudart -lpython3.6m -o /home/guow/jin/CTracker-master/lib/ython_nms/gpu_nms.cpython-36m-x86_64-linux-gnu.so

Answer 3 · 2020-08-28T01:41:02.000Z

Oh, you can copy the nms.py from the dir simpledet/operator_py @jinyl777

Answer 4 · 2020-08-28T02:21:20.000Z

What is your pytorch version？
@jinyl777

Answer 5 · 2020-08-28T02:37:12.000Z

pytorch 0.4.1 py36_py35_py27__9.0.176_7.1.2_2 @pjl1995

Answer 6 · 2020-08-28T02:37:57.000Z

I will try it and thankyou very much @chenrxi

Answer 7 · 2020-08-28T03:05:59.000Z

I copy the simpledet/operator_py/nms.py to lib and modify the line2 .cython.cpu_nms to .cpython_nms .
Then I got No module named 'run_for_track' , well seach the file but do not get the file
Is I miss some file ? @pjl1995

Answer 8 · 2020-08-28T03:37:26.000Z

I am sorry that I forget to modify it. You can just modify 'run_for_track' to 'test' in train.py and it's OK.
@jinyl777

Answer 9 · 2020-08-28T08:55:02.000Z

I modify it and it can run, thanks I will close the issue @pjl1995

Answer 10 · 2020-10-21T08:44:36.000Z

Hi all, I am so sorry that I forgot to add the nms.py in the lib folder. Now this bug is fixed and you can download the latest code.