GrokCV / GrokSAR

Apache License 2.0
7 stars 0 forks source link

GrokSAR

Toolkit for GrokSAR

Installation

Step 1: Create a conda environment

$ conda create --name groksar python=3.9
$ source activate groksar

Step 2: Install PyTorch

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia

Step 3: Install OpenMMLab 2.x Codebases

# 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

Step 4: Install 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

Model Zoo and Benchmark

Note: Both passwords for BaiduYun and OneDrive is grok.

Leaderboard

Model Zoo

DenoDet

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