Wheat_HEAD_Detection_Counting_ComputerVision_Model
/
detectron2
/configs
/PascalVOC-Detection
/faster_rcnn_R_50_FPN.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
MASK_ON: False | |
RESNETS: | |
DEPTH: 50 | |
ROI_HEADS: | |
NUM_CLASSES: 20 | |
INPUT: | |
MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800) | |
MIN_SIZE_TEST: 800 | |
DATASETS: | |
TRAIN: ('voc_2007_trainval', 'voc_2012_trainval') | |
TEST: ('voc_2007_test',) | |
SOLVER: | |
STEPS: (12000, 16000) | |
MAX_ITER: 18000 # 17.4 epochs | |
WARMUP_ITERS: 100 | |