SLA/test.sh at master · ming71/SLA · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#!/usr/bin/env bash

PYTHON=${PYTHON:-'python'}

GPUS='2'

# DATASET='HRSC2016'
# DATASET='UCAS_AOD'
# DATASET='DOTA'
# DATASET='DOTA1.5'
DATASET='DOTA2'

CKPT='50000.pth'


if   [ $DATASET = 'HRSC2016' ]; then
    ROOT='hrsc2016'
    VOC07_METRIC=True

elif [ $DATASET = 'UCAS_AOD' ]; then
    ROOT='ucas_aod'
    VOC07_METRIC=True

elif [ $DATASET = 'DOTA' ]; then
    ROOT='dota'
    VOC07_METRIC=True

elif [ $DATASET = 'DOTA1.5' ]; then
    ROOT='dota1_5'
    VOC07_METRIC=True

elif [ $DATASET = 'DOTA2' ]; then
    ROOT='dota2'
    VOC07_METRIC=True

fi


# python run/$ROOT/evaluate.py --gpus $GPUS --ckpt $CKPT  --use_voc07   $VOC07_METRIC

python -m torch.distributed.launch  run/$ROOT/evaluate_dist.py  --gpus $GPUS --ckpt $CKPT  --use_voc07   $VOC07_METRIC