Win10 下安装 ATSS

1. call VS2019的编译环境

安装Visual Stutio生成工具2019【主要是MSVC，其他不用】
然后 call 自己的 vcvars64.bat，我的vcvars64.bat路径是E:\Microsoft Visual Studio\2019\BuildTools\VC\Auxiliary\Build\vcvars64.bat，故命令如下

call "E:\Microsoft Visual Studio\2019\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
set DISTUTILS_USE_SDK=1

2. 修改源代码

  • 对ATSS\atss_core\csrc\cpu\nms_cpu.cpp、ATSS\atss_core\csrc\cpu\ROIAlign_cpu.cpp进行修改
    #include "cpu/vision.h" 改为 #include "vision.h" ，不然会报错找不到cpu/vision.h文件
  • 将ATSS\atss_core\csrc\cuda\deform_pool_cuda.cu、ATSS\atss_core\csrc\cuda\deform_conv_cuda.cu内的AT_CHECK改为TORCH_CHECK，不然报错 “AT_CHECK” is undefined

3. 安装

直接参照官方提供的安装指南进行安装即可
python setup.py build develop --no-deps

posted @ 2021-07-12 16:01  MorStar  阅读(176)  评论(0)  编辑  收藏  举报