本次安装环境
cuda 10.1 下载地址
pytorch 1.5.1 命令: pip install torch1.5.1+cu101 torchvision0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html 更多安装方式
vs 2017 下载地址
fvcore 0.1.2
下载源码
修改 detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu
1.注释块/* #ifdef WITH_CUDA…#endif*/
2.在后面添加一句 #include “box_iou_rotated/box_iou_rotated_utils.h”
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAApplyUtils.cuh>
//NOTE: replace relative import
/*#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
// TODO avoid this when pytorch supports "same directory" hipification
#ifdef WITH_HIP
#include "box_iou_rotated/box_iou_rotated_utils.h"
#endif*/
#include "box_iou_rotated/box_iou_rotated_utils.h"
编译
进入下载代码的目录执行
"""
@author: qq群686070107
"""
python -m pip install -e detectron2(或者detectron2-master)