| 1 | from detectron2.config import CfgNode as CN |
| 2 | |
| 3 | def add_config(cfg): |
| 4 | |
| 5 | cfg.MODEL.SEM_SEG_HEAD.PIXEL_DECODER_NAME = "MaskDINOEncoder" |
| 6 | |
| 7 | cfg.MODEL.ENCODER = CN() |
| 8 | cfg.MODEL.ENCODER.NUM_CLASSES = 133 |
| 9 | cfg.MODEL.ENCODER.CONVS_DIM = 256 |
| 10 | cfg.MODEL.ENCODER.MASK_DIM = 256 |
| 11 | cfg.MODEL.ENCODER.NORM = "GN" |
| 12 | cfg.MODEL.ENCODER.IN_FEATURES = [ "res2", "res3", "res4", "res5" ] |
| 13 | cfg.MODEL.ENCODER.DEFORMABLE_TRANSFORMER_ENCODER_IN_FEATURES = [ "res2", "res3", "res4", "res5" ] |
| 14 | cfg.MODEL.ENCODER.COMMON_STRIDE = 4 |
| 15 | cfg.MODEL.ENCODER.TRANSFORMER_ENC_LAYERS = 6 |
| 16 | cfg.MODEL.ENCODER.TOTAL_NUM_FEATURE_LEVELS = 5 |
| 17 | cfg.MODEL.ENCODER.NUM_FEATURE_LEVELS = 3 |
| 18 | cfg.MODEL.ENCODER.FEATURE_ORDER = "low2high" |
| 19 | |
| 20 | cfg.MODEL.DECODER = CN() |
| 21 | cfg.MODEL.DECODER.NHEADS = 8 |
| 22 | cfg.MODEL.DECODER.DROPOUT = 0.0 |
| 23 | cfg.MODEL.DECODER.DIM_FEEDFORWARD = 2048 |
| 24 | |
| 25 | cfg.MODEL.TEXT = CN() |
| 26 | cfg.MODEL.TEXT.MODEL_NAME = "" |
| 27 | cfg.MODEL.TEXT.TOKEN_NAME = "" |
| 28 | |
| 29 | cfg.MODEL.TRANSFORMER_DECODER = CN() |
| 30 | cfg.MODEL.TRANSFORMER_DECODER.NAME = "" |
| 31 | cfg.MODEL.TRANSFORMER_DECODER.HIDDEN_DIM = 256 |
| 32 | cfg.MODEL.TRANSFORMER_DECODER.NUM_OBJECT_QUERIES = 900 |
| 33 | cfg.MODEL.TRANSFORMER_DECODER.NHEADS = 8 |
| 34 | cfg.MODEL.TRANSFORMER_DECODER.DROPOUT = 0.0 |
| 35 | cfg.MODEL.TRANSFORMER_DECODER.DIM_FEEDFORWARD = 2048 |
| 36 | cfg.MODEL.TRANSFORMER_DECODER.DEC_LAYERS = 9 |
| 37 | cfg.MODEL.TRANSFORMER_DECODER.TOTAL_NUM_FEATURE_LEVELS = 5 |
| 38 | cfg.MODEL.TRANSFORMER_DECODER.MASK_DIM = 256 |
| 39 | cfg.MODEL.TRANSFORMER_DECODER.LEARN_TGT = False |
| 40 | |
| 41 | cfg.MODEL.SAM_WEIGHTS = "" |