mirror of https://github.com/JosephKJ/OWOD.git
20 lines
471 B
YAML
20 lines
471 B
YAML
|
_BASE_: "../Base-RCNN-FPN.yaml"
|
||
|
|
MODEL:
|
||
|
|
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
||
|
|
MASK_ON: True
|
||
|
|
RESNETS:
|
||
|
|
DEPTH: 50
|
||
|
|
ROI_HEADS:
|
||
|
|
NUM_CLASSES: 1230
|
||
|
|
SCORE_THRESH_TEST: 0.0001
|
||
|
|
INPUT:
|
||
|
|
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
|
||
|
|
DATASETS:
|
||
|
|
TRAIN: ("lvis_v0.5_train",)
|
||
|
|
TEST: ("lvis_v0.5_val",)
|
||
|
|
TEST:
|
||
|
|
DETECTIONS_PER_IMAGE: 300 # LVIS allows up to 300
|
||
|
|
DATALOADER:
|
||
|
|
SAMPLER_TRAIN: "RepeatFactorTrainingSampler"
|
||
|
|
REPEAT_THRESHOLD: 0.001
|