Wheat_HEAD_Detection_Counting_ComputerVision_Model
/
detectron2
/configs
/LVISv1-InstanceSegmentation
/mask_rcnn_R_50_FPN_1x.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
MASK_ON: True | |
RESNETS: | |
DEPTH: 50 | |
ROI_HEADS: | |
NUM_CLASSES: 1203 | |
SCORE_THRESH_TEST: 0.0001 | |
INPUT: | |
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800) | |
DATASETS: | |
TRAIN: ("lvis_v1_train",) | |
TEST: ("lvis_v1_val",) | |
TEST: | |
DETECTIONS_PER_IMAGE: 300 # LVIS allows up to 300 | |
SOLVER: | |
STEPS: (120000, 160000) | |
MAX_ITER: 180000 # 180000 * 16 / 100000 ~ 28.8 epochs | |
DATALOADER: | |
SAMPLER_TRAIN: "RepeatFactorTrainingSampler" | |
REPEAT_THRESHOLD: 0.001 | |