search

endernewton / tf-faster-rcnn

Tensorflow Faster RCNN for Object Detection
https://arxiv.org/pdf/1702.02138.pdf
MIT License
3.65k stars 1.58k forks source link

train error #466

Open wudi00 opened 4 years ago

wudi00 commented 4 years ago

When I train my own datasets, there is such error: Traceback (most recent call last): File "./tools/trainval_net.py", line 11, in <module> from lib.model.train_val import get_training_roidb, train_net ModuleNotFoundError: No module named 'lib' I have make the lib as following, but it doesn't work. make clean make cd .. The log is `building 'nms.gpu_nms' extension {'gcc': ['-Wno-unused-function'], 'nvcc': ['-arch=sm_61', '--ptxas-options=-v', '-c', '--compiler-options', "'-fPIC'"]} /usr/local/cuda/bin/nvcc -I/media/disk/wudi/.local/lib/python3.6/site-packages/numpy/core/include -I/usr/local/cuda/include -I/opt/anaconda3/envs/py36/include/python3.6m -c nms/nms_kernel.cu -o build/temp.linux-x86_64-3.6/nms/nms_kernel.o -arch=sm_61 --ptxas-options=-v -c --compiler-options '-fPIC' ptxas info : 0 bytes gmem ptxas info : Compiling entry function '_Z10nms_kernelifPKfPy' for 'sm_61' ptxas info : Function properties for _Z10nms_kernelifPKfPy 0 bytes stack frame, 0 bytes spill stores, 0 bytes spill loads ptxas info : Used 25 registers, 1280 bytes smem, 344 bytes cmem[0], 12 bytes cmem[2] {'gcc': ['-Wno-unused-function'], 'nvcc': ['-arch=sm_61', '--ptxas-options=-v', '-c', '--compiler-options', "'-fPIC'"]} gcc -pthread -B /opt/anaconda3/envs/py36/compiler_compat -Wl,--sysroot=/ -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -I/media/disk/wudi/.local/lib/python3.6/site-packages/numpy/core/include -I/usr/local/cuda/include -I/opt/anaconda3/envs/py36/include/python3.6m -c nms/gpu_nms.cpp -o build/temp.linux-x86_64-3.6/nms/gpu_nms.o -Wno-unused-function cc1plus: warning: command line option '-Wstrict-prototypes' is valid for C/ObjC but not for C++ In file included from /media/disk/wudi/.local/lib/python3.6/site-packages/numpy/core/include/numpy/ndarraytypes.h:1821:0, from /media/disk/wudi/.local/lib/python3.6/site-packages/numpy/core/include/numpy/ndarrayobject.h:18, from /media/disk/wudi/.local/lib/python3.6/site-packages/numpy/core/include/numpy/arrayobject.h:4, from nms/gpu_nms.cpp:346: /media/disk/wudi/.local/lib/python3.6/site-packages/numpy/core/include/numpy/npy_1_7_deprecated_api.h:15:2: warning: #warning "Using deprecated NumPy API, disable it by " "#defining NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION" [-Wcpp]

warning "Using deprecated NumPy API, disable it by " \

^~~ g++ -pthread -shared -B /opt/anaconda3/envs/py36/compiler_compat -L/opt/anaconda3/envs/py36/lib -Wl,-rpath=/opt/anaconda3/envs/py36/lib -Wl,--no-as-needed -Wl,--sysroot=/ build/temp.linux-x86_64-3.6/nms/nms_kernel.o build/temp.linux-x86_64-3.6/nms/gpu_nms.o -L/usr/local/cuda/lib64 -Wl,-R/usr/local/cuda/lib64 -lcudart -o /media/disk/wudi/tf-faster-rcnn/lib/nms/gpu_nms.cpython-36m-x86_64-linux-gnu.so rm -rf build`

wudi00 commented 4 years ago

It has solved.

snant commented 4 years ago

hey,bro!i want to know that how to operate the project.I feel like i don't know how to relize it.please give me a little advice.thank you very much!