File size: 957 Bytes
c985ba4 |
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 |
class DefaultModelConfig():
def __init__(self):
self.MODEL_NAME = 'AOTDefault'
self.MODEL_VOS = 'aot'
self.MODEL_ENGINE = 'aotengine'
self.MODEL_ALIGN_CORNERS = True
self.MODEL_ENCODER = 'mobilenetv2'
self.MODEL_ENCODER_PRETRAIN = './pretrain_models/mobilenet_v2-b0353104.pth'
self.MODEL_ENCODER_DIM = [24, 32, 96, 1280] # 4x, 8x, 16x, 16x
self.MODEL_ENCODER_EMBEDDING_DIM = 256
self.MODEL_DECODER_INTERMEDIATE_LSTT = True
self.MODEL_FREEZE_BN = True
self.MODEL_FREEZE_BACKBONE = False
self.MODEL_MAX_OBJ_NUM = 10
self.MODEL_SELF_HEADS = 8
self.MODEL_ATT_HEADS = 8
self.MODEL_LSTT_NUM = 1
self.MODEL_EPSILON = 1e-5
self.MODEL_USE_PREV_PROB = False
self.TRAIN_LONG_TERM_MEM_GAP = 9999
self.TRAIN_AUG_TYPE = 'v1'
self.TEST_LONG_TERM_MEM_GAP = 9999
self.TEST_SHORT_TERM_MEM_SKIP = 1
|