diff options
author | Manu Mathew | 2020-06-23 23:02:47 -0500 |
---|---|---|
committer | Manu Mathew | 2020-06-24 02:26:24 -0500 |
commit | 4dadffe68d14141f17abcc48cc7d3c046944d81a (patch) | |
tree | 22cda9be5bf1cdc2b75b18e8b0a49d9085ce49ba /scripts/test_main.py | |
parent | baeb92b79c56105b4e4d2980966729d359f2cb38 (diff) | |
download | pytorch-mmdetection-4dadffe68d14141f17abcc48cc7d3c046944d81a.tar.gz pytorch-mmdetection-4dadffe68d14141f17abcc48cc7d3c046944d81a.tar.xz pytorch-mmdetection-4dadffe68d14141f17abcc48cc7d3c046944d81a.zip |
bipfn and scripts updated
Diffstat (limited to 'scripts/test_main.py')
-rwxr-xr-x | scripts/test_main.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/scripts/test_main.py b/scripts/test_main.py index 71c53394..091e8086 100755 --- a/scripts/test_main.py +++ b/scripts/test_main.py | |||
@@ -20,8 +20,9 @@ config='./configs/ssd/ssd_mobilenet_fpn.py' | |||
20 | config='./configs/ssd/ssd_resnet_fpn.py' | 20 | config='./configs/ssd/ssd_resnet_fpn.py' |
21 | config='./configs/ssd/ssd_regnet_fpn_bgr.py' | 21 | config='./configs/ssd/ssd_regnet_fpn_bgr.py' |
22 | 22 | ||
23 | config='./configs/retinanet/retinanet_regnet_fpn_bgr.py' | ||
24 | config='./configs/retinanet/retinanet_resnet_fpn.py' | 23 | config='./configs/retinanet/retinanet_resnet_fpn.py' |
24 | config='./configs/retinanet/retinanet_regnet_fpn_bgr.py' | ||
25 | config='./configs/retinanet/retinanet_regnet_bifpn_bgr.py' | ||
25 | config='./configs/retinanet/fcos_regnet_fpn_bgr.py' | 26 | config='./configs/retinanet/fcos_regnet_fpn_bgr.py' |
26 | ''' | 27 | ''' |
27 | 28 | ||
@@ -30,7 +31,7 @@ config='./configs/retinanet/retinanet_regnet_fpn_bgr.py' | |||
30 | ######################################################################## | 31 | ######################################################################## |
31 | # other settings | 32 | # other settings |
32 | gpus = 4 | 33 | gpus = 4 |
33 | dataset_style = 'voc' #'voc' #'coco' | 34 | dataset_style = 'coco' #'voc' #'coco' |
34 | 35 | ||
35 | 36 | ||
36 | ######################################################################### | 37 | ######################################################################### |