MaskRCNN/pytorch/configs/nd_baselines/e2e_mask_rcnn_R_50_FPN_1x_nd.yaml [1:47]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
MODEL:
  META_ARCHITECTURE: "GeneralizedRCNN"
  WEIGHT: "/home/ubuntu/efs/chengxi/pretrained/resnet50d_blk64.pth"
  BACKBONE:
    CONV_BODY: "R-50-FPN"
    OUT_CHANNELS: 256
    FREEZE_CONV_BODY_AT: -1
  RESNETS: # use DECONV for backbone
    TRANS_FUNC: "BottleneckWithDeconv"
    STEM_FUNC: "StemWithDeconv"
  FPN:
    USE_DECONV: True # use ND for FPN
  RPN:
    USE_FPN: True
    ANCHOR_STRIDE: (4, 8, 16, 32, 64)
    PRE_NMS_TOP_N_TRAIN: 2000
    PRE_NMS_TOP_N_TEST: 1000
    POST_NMS_TOP_N_TEST: 1000
    FPN_POST_NMS_TOP_N_TEST: 1000
    FPN_POST_NMS_TOP_N_TRAIN: 4000
    USE_DECONV: True # use ND for RPN

  ROI_HEADS:
    USE_FPN: True
  ROI_BOX_HEAD:
    USE_DECONV: True # use ND for bbox head
    POOLER_RESOLUTION: 7
    POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
    POOLER_SAMPLING_RATIO: 2
    FEATURE_EXTRACTOR: "FPN2MLPFeatureExtractor"
    PREDICTOR: "FPNPredictor"
  ROI_MASK_HEAD:
    USE_DECONV: True # use ND for mask head
    POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
    FEATURE_EXTRACTOR: "MaskRCNNFPNFeatureExtractor"
    PREDICTOR: "MaskRCNNC4Predictor"
    POOLER_RESOLUTION: 14
    POOLER_SAMPLING_RATIO: 2
    RESOLUTION: 28
    SHARE_BOX_FEATURE_EXTRACTOR: False
  MASK_ON: True
  DECONV: 
    BLOCK: 64
    BLOCK_FC: 256
    SYNC: True
    #ITER: 5
    STRIDE: 5
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



MaskRCNN/pytorch/configs/nd_baselines/e2e_mask_rcnn_R_50_FPN_1x_nd_bs128.yaml [1:47]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
MODEL:
  META_ARCHITECTURE: "GeneralizedRCNN"
  WEIGHT: "/home/ubuntu/efs/chengxi/pretrained/resnet50d_blk64.pth"
  BACKBONE:
    CONV_BODY: "R-50-FPN"
    OUT_CHANNELS: 256
    FREEZE_CONV_BODY_AT: -1
  RESNETS: # use DECONV for backbone
    TRANS_FUNC: "BottleneckWithDeconv"
    STEM_FUNC: "StemWithDeconv"
  FPN:
    USE_DECONV: True # use ND for FPN
  RPN:
    USE_FPN: True
    ANCHOR_STRIDE: (4, 8, 16, 32, 64)
    PRE_NMS_TOP_N_TRAIN: 2000
    PRE_NMS_TOP_N_TEST: 1000
    POST_NMS_TOP_N_TEST: 1000
    FPN_POST_NMS_TOP_N_TEST: 1000
    FPN_POST_NMS_TOP_N_TRAIN: 4000
    USE_DECONV: True # use ND for RPN

  ROI_HEADS:
    USE_FPN: True
  ROI_BOX_HEAD:
    USE_DECONV: True # use ND for bbox head
    POOLER_RESOLUTION: 7
    POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
    POOLER_SAMPLING_RATIO: 2
    FEATURE_EXTRACTOR: "FPN2MLPFeatureExtractor"
    PREDICTOR: "FPNPredictor"
  ROI_MASK_HEAD:
    USE_DECONV: True # use ND for mask head
    POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
    FEATURE_EXTRACTOR: "MaskRCNNFPNFeatureExtractor"
    PREDICTOR: "MaskRCNNC4Predictor"
    POOLER_RESOLUTION: 14
    POOLER_SAMPLING_RATIO: 2
    RESOLUTION: 28
    SHARE_BOX_FEATURE_EXTRACTOR: False
  MASK_ON: True
  DECONV: 
    BLOCK: 64
    BLOCK_FC: 256
    SYNC: True
    #ITER: 5
    STRIDE: 5
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



