安装detectron2的一些坑

想安装detectron2这个目标检测框架,竟然不能直接用pip安装。

运行如下命令:

git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2

但是却出了问题:

问题就是ninja -v 出错,安装了ninja后还是出错,在本地目录下运行没问题,出错:
subprocess.CalledProcessError: Command ‘[‘ninja’, ‘-v’]’ returned non-zero exit status 1

开始以为是运行 ninja -v出现的问题,最后花费了很大功夫还是没解决。
最后干脆把ninja关闭了。
就是把detectron2目录下setup.py最后一项更改如下:


    cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension.with_options(use_ninja=False)},

这样就不适用ninja来编译了。

detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu
开始是说
using namespace detectron2;
这句不认识命名空间detectron2
就把这句注释掉了
但是接着说不识别函数:single_box_iou_rotated。
最后发现这个函数在detectron2\detectron2\layers\csrc\box_iou_rotated\box_iou_rotated_utils.h
中存在。

最后查找资料,要更改nms_rotated_cuda.cu文件中如下:


// TODO avoid this when pytorch supports "same directory" hipification

*/

using namespace detectron2;

就是把两个对宏判断去掉:
WITH_CUDA和 WITH_HIP
可能编译的时候没打开这两个宏开关。
最后运行成功了
python -m pip install -e detectron2

这个问题花费了半天才解决。
所以环境搭建这样的问题,确实比较麻烦,有时比调程序找bug麻烦。

Original: https://blog.csdn.net/jiankangyq/article/details/121879936
Author: 平原君2088
Title: 安装detectron2的一些坑

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/682950/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球