VideoCrafterXtend
/
VBench
/vbench
/third_party
/grit_src
/centernet2
/configs
/Base-CenterNet-FPN.yaml
MODEL: | |
META_ARCHITECTURE: "CenterNetDetector" | |
PROPOSAL_GENERATOR: | |
NAME: "CenterNet" | |
BACKBONE: | |
NAME: "build_p67_resnet_fpn_backbone" | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
RESNETS: | |
DEPTH: 50 | |
OUT_FEATURES: ["res3", "res4", "res5"] | |
FPN: | |
IN_FEATURES: ["res3", "res4", "res5"] | |
DATASETS: | |
TRAIN: ("coco_2017_train",) | |
TEST: ("coco_2017_val",) | |
SOLVER: | |
IMS_PER_BATCH: 16 | |
BASE_LR: 0.01 | |
STEPS: (60000, 80000) | |
MAX_ITER: 90000 | |
CHECKPOINT_PERIOD: 1000000000 | |
WARMUP_ITERS: 4000 | |
WARMUP_FACTOR: 0.00025 | |
CLIP_GRADIENTS: | |
ENABLED: True | |
INPUT: | |
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800) | |
OUTPUT_DIR: "./output/CenterNet2/auto" | |