Anaconda中安装Cascade RCNN(Detectron)的若干问题

安装参考https://github.com/zhaoweicai/Detectron-Cascade-RCNN/blob/master/INSTALL.md

1、对于在 python detectron/tests/test_spatial_narrow_as_op.py 时出现 AssertionError: Detectron ops lib not found; make sure that your Caffe2 version includes Detectron module

detectron/utils/env.py中第65行左右的:

ops_path = os.path.join(prefix, 'lib/libcaffe2_detectron_ops_gpu.so')

在conda的envs文件夹中搜索这个so,将整个get_detectron_ops_lib()的函数内容替换为 return "/home/wit/anaconda3/envs/xxx/lib/python2.7/site-packages/torch/lib/libcaffe2_detectron_ops_gpu.so"。

猜你喜欢

转载自www.cnblogs.com/aimhabo/p/11471315.html