MODEL: META_ARCHITECTURE: "GeneralizedRCNN" RPN: PRE_NMS_TOPK_TEST: 6000 POST_NMS_TOPK_TEST: 1000 ROI_HEADS: # NUM_CLASSES: 81 # 0-79 Known class; 80 -> Unknown; 81 -> Background. NUM_CLASSES: 81 NAME: "Res5ROIHeads" WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" MASK_ON: False RESNETS: DEPTH: 50 INPUT: MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800) MIN_SIZE_TEST: 800 DATASETS: TRAIN: ("coco_2017_train",) TEST: ("coco_2017_val",) SOLVER: IMS_PER_BATCH: 16 BASE_LR: 0.02 STEPS: (60000, 80000) MAX_ITER: 90000 VERSION: 2 OWOD: ENABLE_THRESHOLD_AUTOLABEL_UNK: True NUM_UNK_PER_IMAGE: 1 ENABLE_UNCERTAINITY_AUTOLABEL_UNK: False ENABLE_CLUSTERING: True FEATURE_STORE_SAVE_PATH: 'feature_store' CLUSTERING: ITEMS_PER_CLASS: 20 START_ITER: 0 #10000 UPDATE_MU_ITER: 1 MOMENTUM: 0.99 Z_DIMENSION: 128 UNK_THRESH: 0.7 CALIBRATION: 0.7 GENERATE_CALI: False