Toolkit for GrokSAR
$ conda create --name groksar python=3.9
$ source activate groksar
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
# openmmlab codebases
pip install -U openmim dadaptation --no-input
mim install mmengine "mmcv>=2.0.0" "mmdet>=3.0.0" "mmsegmentation>=1.0.0" "mmrotate>=1.0.0rc1" mmyolo
# heatmap generation dependencies
pip install grad-cam
# other dependencies
pip install ninja --no-input
pip install scikit-learn
pip install psutil
groksar
python setup.py develop
Note: make sure you have cd
to the root directory of groksar
$ git clone git@github.com:GrokCV/groksar.git
$ cd groksar
Note: Both passwords for BaiduYun and OneDrive is grok
.
MSAR
Model | mAP(07) | mAP(12) | FLOPs | Config | Training Log | Checkpoint |
DenoDet | 69.90 | 71.21 | 12.89G | DenoDet_3x_MSAR.py | 百度网盘 | OneDirve |
SAR-AIRcraft-1.0
Model | mAP(07) | mAP(12) | FLOPs | Config | Training Log | Checkpoint |
DenoDet | 68.60 | 69.56 | 48.53G | DenoDet_1x_SAR-AIRcraft-1.0.py | 百度网盘 | OneDirve |
AIR-SARShip-1.0
Model | mAP(07) | mAP(12) | FLOPs | Config | Training Log | Checkpoint |
DenoDet | 72.42 | 73.36 | 48.52G | DenoDet_6x_AIR-SARShip-1.0.py | 百度网盘 | OneDirve |