VLog / models /grit_src /third_party /CenterNet2 /configs /quick_schedules /keypoint_rcnn_R_50_FPN_instant_test.yaml
leiwx52's picture
VLog hf gradio demo
5a444be
raw
history blame
346 Bytes
_BASE_: "../Base-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
KEYPOINT_ON: True
ROI_HEADS:
NUM_CLASSES: 1
DATASETS:
TRAIN: ("keypoints_coco_2017_val_100",)
TEST: ("keypoints_coco_2017_val_100",)
SOLVER:
BASE_LR: 0.005
STEPS: (30,)
MAX_ITER: 40
IMS_PER_BATCH: 4
DATALOADER:
NUM_WORKERS: 2