论文官方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
不行,版本太高了找不到解决方法,还是老老实实降低版本吧。