File size: 655 Bytes
b94fa0f |
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 |
_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 |