复现失败S2ANet :Align Deep Features for Oriented Object Detection

论文官方GITHUB
https://github.com/csuhan/s2anet
这篇文章被TGRS接收。
2022年4月6日16点00分

0.租GPU

GPU租用AUTODL的 RTX 2080Ti

1.install.md

这次直接使用autodl自带的pytorch试试，
PyTorch 1.6.0
Python 3.8
Cuda 10.1

conda create -n s2anet python=3.7 -y
source activate s2anet

下一步clone仓库卡住了，睡大觉。

git clone https://github.com/csuhan/s2anet.git
cd s2anet

2022年4月6日16点46分
2022年4月6日17点48分
发现chrome浏览器对jupty支持不友好，老卡，换了edge好多了。

cd s2anet
pip install -r requirements.txt
python setup.py develop

安装完了。
运行python setup.py develop时候现式nomodul named pytorch。我意识到，从autodl上租用GPU配套的pytorch和python会在base环境下使用。但是使用conda创建新环境之后就不能用了，需要在conda中再次配置pytorch。
在conda新创的s2anet环境中安装

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

Install DOTA_devkit

conda install swig
cd DOTA_devkit/polyiou
swig -c++ -python csrc/polyiou.i
python setup.py build_ext --inplace

然后准备数据集。

2.start

python tools/test.py configs/hrsc2016/s2anet_r101_fpn_3x_hrsc2016.py work_dirs/s2anet_r101_fpn_3x_hrsc2016_converted-4a4548e1.pth --out work_dirs/s2anet_r50_fpn_3x_hrsc2016/res.pkl --data hrsc2016

结果报错说没有mmdet框架
把之前mmcv或者mmcv-full卸载

pip uninstall mmcv
pip uninstall mmcv-full

使用mim安装mmdet，会自动把缺少的安装的，但是多了很难匹配版本。

pip install mim
mim install mmdet

运行报错

ImportError: cannot import name 'init_dist' from 'mmdet.apis'

应该是GPU和cpu版本的问题，有个相关的错误参考
https://github.com/csuhan/s2anet/issues/48
，刚才是无卡模式运行的，这次带着GPU试试。
先运行

python setup.py develop

好家伙，用GPU得编译小二十分钟来。
还是报错，把test.py的

from mmdet.apis import init_dist
改成
from mmcv.runner import init_dist

不行，版本太高了找不到解决方法，还是老老实实降低版本吧。