Spaces:
Sleeping
Sleeping
| _BASE_: "Base-RCNN-FPN.yaml" | |
| MODEL: | |
| META_ARCHITECTURE: "GeneralizedRCNN" | |
| WEIGHTS: "model_final.pth" | |
| #WEIGHTS: "./data/SPEED-Detection/faster-rcnn/vanilla/random_seed_0/model_final.pth" | |
| MASK_ON: False | |
| RESNETS: | |
| DEPTH: 50 | |
| ROI_HEADS: | |
| NAME: "StandardROIHeads" | |
| NUM_CLASSES: 9 | |
| # INPUT: | |
| # MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800) | |
| # MIN_SIZE_TEST: 800 | |
| DATASETS: | |
| TRAIN: ('speed_custom_train',) | |
| TEST: ('speed_custom_val',) | |
| SOLVER: | |
| IMS_PER_BATCH: 18 | |
| BASE_LR: 0.01 | |
| STEPS: (2800, 3700) | |
| MAX_ITER: 4200 | |
| WARMUP_ITERS: 50 | |
| CHECKPOINT_PERIOD : 2100 | |
| TEST: | |
| EVAL_PERIOD: 1000 | |