|
MODEL: |
|
META_ARCHITECTURE: "GeneralizedRCNN" |
|
RESNETS: |
|
OUT_FEATURES: ["res5"] |
|
RES5_DILATION: 2 |
|
RPN: |
|
IN_FEATURES: ["res5"] |
|
PRE_NMS_TOPK_TEST: 6000 |
|
POST_NMS_TOPK_TEST: 1000 |
|
ROI_HEADS: |
|
NAME: "StandardROIHeads" |
|
IN_FEATURES: ["res5"] |
|
ROI_BOX_HEAD: |
|
NAME: "FastRCNNConvFCHead" |
|
NUM_FC: 2 |
|
POOLER_RESOLUTION: 7 |
|
ROI_MASK_HEAD: |
|
NAME: "MaskRCNNConvUpsampleHead" |
|
NUM_CONV: 4 |
|
POOLER_RESOLUTION: 14 |
|
DATASETS: |
|
TRAIN: ("coco_2017_train",) |
|
TEST: ("coco_2017_val",) |
|
SOLVER: |
|
IMS_PER_BATCH: 16 |
|
BASE_LR: 0.02 |
|
STEPS: (60000, 80000) |
|
MAX_ITER: 90000 |
|
INPUT: |
|
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800) |
|
VERSION: 2 |
|
|