Faster-RCNN Tensorflow No module named ‘libs.box_utils.cython_utils.cython_bbox‘ 错误解决方法

原创于 2021-10-03 23:26:52 发布 · 1.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #cv #目标跟踪

机器学习同时被 2 个专栏收录

6 篇文章

订阅专栏

python

2 篇文章

订阅专栏

这篇博客介绍了在Python项目中遇到的Cython模块导入错误，如'libs.box_utils.cython_utils.cython_bbox'和'libs.box_utils.cython_utils.cython_nms'。通过进入相应目录并运行`python setup.py build_ext --inplace`，在未安装CUDA的情况下注释掉CUDA配置相关代码，解决了编译问题。在完成编译后，项目能够正常运行。此外，还提到了如果需要运行libs/setup.py，需要同时注释掉与CUDA相关的内容。

部署运行你感兴趣的模型镜像

运行时报错：

ImportError: No module named 'libs.box_utils.cython_utils.cython_bbox' 或

ImportError: No module named 'libs.box_utils.cython_utils.cython_nms'

需要先进入 libs/box_utils/cython_utils，

运行 python setup.py build_ext --inplace，显示

Traceback (most recent call last):
File "setup.py", line 55, in <module>
CUDA = locate_cuda()
... ...
AttributeError: 'dict' object has no attribute 'iteritems'

此处需要对cuda进行一些配置，如果暂未安装cuda，可以先将55行： CUDA = locate_cuda() 注释掉，之后再运行 python setup.py build_ext --inplace，此时显示：

{'gcc': ['-Wno-cpp', '-Wno-unused-function']}
gcc -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -m64 -fPIC -fPIC -I/home/lishixiang/anaconda2/envs/py3.5/lib/python3.5/site-packages/numpy/core/include -I/home/lishixiang/anaconda2/envs/py3.5/include/python3.5m -c bbox.c -o build/temp.linux-x86_64-3.5/bbox.o -Wno-cpp -Wno-unused-function
creating build/lib.linux-x86_64-3.5
gcc -pthread -shared -L/home/lishixiang/anaconda2/envs/py3.5/lib -Wl,-rpath=/home/lishixiang/anaconda2/envs/py3.5/lib,--no-as-needed build/temp.linux-x86_64-3.5/bbox.o -L/home/lishixiang/anaconda2/envs/py3.5/lib -lpython3.5m -o build/lib.linux-x86_64-3.5/cython_bbox.cpython-35m-x86_64-linux-gnu.so
skipping 'nms.c' Cython extension (up-to-date)
building 'cython_nms' extension
{'gcc': ['-Wno-cpp', '-Wno-unused-function']}
gcc -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -m64 -fPIC -fPIC -I/home/lishixiang/anaconda2/envs/py3.5/lib/python3.5/site-packages/numpy/core/include -I/home/lishixiang/anaconda2/envs/py3.5/include/python3.5m -c nms.c -o build/temp.linux-x86_64-3.5/nms.o -Wno-cpp -Wno-unused-function
gcc -pthread -shared -L/home/lishixiang/anaconda2/envs/py3.5/lib -Wl,-rpath=/home/lishixiang/anaconda2/envs/py3.5/lib,--no-as-needed build/temp.linux-x86_64-3.5/nms.o -L/home/lishixiang/anaconda2/envs/py3.5/lib -lpython3.5m -o build/lib.linux-x86_64-3.5/cython_nms.cpython-35m-x86_64-linux-gnu.so

安装成功！

此时再运行相关程序，可以正确运行。

补充：如需运行 libs/setup.py，除了需要将第59行：CUDA = locate_cuda() 注释掉，还需要将以下的176-192行与cuda相关的内容注释掉，方能运行成功：

Extension('rotation.rbbox_overlaps',
    ['rotation/rbbox_overlaps_kernel.cu', 'rotation/rbbox_overlaps.pyx'],
    library_dirs=[CUDA['lib64']],
    libraries=['cudart'],
    language='c++',
    runtime_library_dirs=[CUDA['lib64']],
    # this syntax is specific to this build system
    # we're only going to use certain compiler args with nvcc and not with
    # gcc the implementation of this trick is in customize_compiler() below
    extra_compile_args={'gcc': ["-Wno-unused-function"],
                        'nvcc': ['-arch=sm_35',
                                 '--ptxas-options=-v',
                                 '-c',
                                 '--compiler-options',
                                 "'-fPIC'"]},
    include_dirs = [numpy_include, CUDA['include']]
),

您可能感兴趣的与本文相关的镜像