安装detercton2

ruixia.chen

于 2024-05-31 17:21:34 发布

阅读量108

点赞数 3

文章标签： pip conda python

本文链接：https://blog.csdn.net/u010674979/article/details/139357943

版权

# Windows 从源码安装
下载detectorn2到本地, 注意detectorn2的版本，一定是下载v0.6的版本！！！
git clone -b v0.6 https://github.com/facebookresearch/detectron2
cd detectron2
python setup.py build --force develop

windows端，如果编译出错，需要将`detectron2/detectron2/layers/csrc/nms_rotated/nms_rotated_cuda.cu`文件中第6-8行:
#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
改为：
#include "box_iou_rotated/box_iou_rotated_utils.h"

# 验证-目标检测
python demo/demo.py --config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml --input jy.jpg --opts MODEL.WEIGHTS 'model_final_b275ba.pkl'
# pkl文件下载地址：https://github.com/facebookresearch/detectron2/blob/main/MODEL_ZOO.md

# Linux (官方下载地址：https://detectron2.readthedocs.io/en/latest/tutorials/install.html)
python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.9/index.html