Spaces:
Sleeping
Sleeping
File size: 1,455 Bytes
a153c95 |
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 |
from detectron2.config import CfgNode as CN
def add_regionspot_config(cfg):
"""
Add config for RegionSpot
"""
cfg.MODEL.RegionSpot = CN()
cfg.MODEL.CLIP_TYPE = 'CLIP_400M_Large'
cfg.MODEL.CLIP_INPUT_SIZE = 224
# Inference
cfg.MODEL.TRAINING = True
cfg.MODEL.BOX_TYPE = 'GT'
#Dataloder
cfg.DATALOADER.DATASET_RATIO = [1,1,1] # sample ratio
cfg.DATALOADER.USE_RFS = [False, False, False]
cfg.DATALOADER.MULTI_DATASET_GROUPING = True # Always true when multi-dataset is enabled
cfg.DATALOADER.DATASET_ANN = ['box', 'box', 'box'] # Annotation type of each dataset
cfg.DATALOADER.USE_DIFF_BS_SIZE = False # Use different batchsize for each dataset
cfg.DATALOADER.DATASET_BS = [8, 32] # Used when USE_DIFF_BS_SIZE is on
# Optimizer.
cfg.SOLVER.OPTIMIZER = "ADAMW"
cfg.SOLVER.BACKBONE_MULTIPLIER = 1.0
# TTA.
cfg.TEST.AUG.MIN_SIZES = (400, 500, 600, 640, 700, 900, 1000, 1100, 1200, 1300, 1400, 1800, 800)
cfg.TEST.AUG.CVPODS_TTA = True
cfg.TEST.AUG.SCALE_FILTER = True
cfg.TEST.AUG.SCALE_RANGES = ([96, 10000], [96, 10000],
[64, 10000], [64, 10000],
[64, 10000], [0, 10000],
[0, 10000], [0, 256],
[0, 256], [0, 192],
[0, 192], [0, 96],
[0, 10000])
|