Add at new repo again

This commit is contained in:
2025-01-28 21:48:35 +00:00
commit 6e660ddb3c
564 changed files with 75575 additions and 0 deletions

View File

@@ -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

View File

@@ -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

View File

@@ -0,0 +1,6 @@
_BASE_: "Base-TridentNet-Fast-C4.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
MASK_ON: False
RESNETS:
DEPTH: 50

View File

@@ -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