Add at new repo again
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
MODEL:
|
||||
META_ARCHITECTURE: "GeneralizedRCNN"
|
||||
BACKBONE:
|
||||
NAME: "build_trident_resnet_backbone"
|
||||
ROI_HEADS:
|
||||
NAME: "TridentRes5ROIHeads"
|
||||
POSITIVE_FRACTION: 0.5
|
||||
BATCH_SIZE_PER_IMAGE: 128
|
||||
PROPOSAL_APPEND_GT: False
|
||||
PROPOSAL_GENERATOR:
|
||||
NAME: "TridentRPN"
|
||||
RPN:
|
||||
POST_NMS_TOPK_TRAIN: 500
|
||||
TRIDENT:
|
||||
NUM_BRANCH: 3
|
||||
BRANCH_DILATIONS: [1, 2, 3]
|
||||
TEST_BRANCH_IDX: 1
|
||||
TRIDENT_STAGE: "res4"
|
||||
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
|
@@ -0,0 +1,9 @@
|
||||
_BASE_: "Base-TridentNet-Fast-C4.yaml"
|
||||
MODEL:
|
||||
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
||||
MASK_ON: False
|
||||
RESNETS:
|
||||
DEPTH: 101
|
||||
SOLVER:
|
||||
STEPS: (210000, 250000)
|
||||
MAX_ITER: 270000
|
@@ -0,0 +1,6 @@
|
||||
_BASE_: "Base-TridentNet-Fast-C4.yaml"
|
||||
MODEL:
|
||||
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
||||
MASK_ON: False
|
||||
RESNETS:
|
||||
DEPTH: 50
|
@@ -0,0 +1,9 @@
|
||||
_BASE_: "Base-TridentNet-Fast-C4.yaml"
|
||||
MODEL:
|
||||
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
||||
MASK_ON: False
|
||||
RESNETS:
|
||||
DEPTH: 50
|
||||
SOLVER:
|
||||
STEPS: (210000, 250000)
|
||||
MAX_ITER: 270000
|
Reference in New Issue
Block a user