| _BASE_: "./Base-RetinaNet.yaml" | |
| MODEL: | |
| META_ARCHITECTURE: "OpenSetRetinaNet" | |
| WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
| RESNETS: | |
| DEPTH: 50 | |
| RETINANET: | |
| NUM_CLASSES: 81 | |
| NUM_KNOWN_CLASSES: 20 | |
| DATASETS: | |
| TRAIN: ('voc_2007_train', 'voc_2012_trainval') | |
| TEST: ('voc_2007_test', 'voc_coco_20_40_test', 'voc_coco_20_60_test', 'voc_coco_20_80_test', 'voc_coco_2500_test', 'voc_coco_5000_test', 'voc_coco_10000_test', 'voc_coco_20000_test') | |
| SOLVER: | |
| STEPS: (21000, 29000) | |
| MAX_ITER: 32000 | |
| WARMUP_ITERS: 1000 | |
| AMP: | |
| ENABLED: True | |
| UPLOSS: | |
| TOPK: 10 | |
| WEIGHT: 0.2 | |
| ICLOSS: | |
| QUEUE_SIZE: 1024 | |
| IN_QUEUE_SIZE: 64 | |
| WEIGHT: 0.2 |