18 lines
449 B
YAML
18 lines
449 B
YAML
_BASE_: "../Base-RCNN-FPN.yaml"
|
|
MODEL:
|
|
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
|
MASK_ON: False
|
|
RESNETS:
|
|
DEPTH: 50
|
|
# Detectron1 uses smooth L1 loss with some magic beta values.
|
|
# The defaults are changed to L1 loss in Detectron2.
|
|
RPN:
|
|
SMOOTH_L1_BETA: 0.1111
|
|
ROI_BOX_HEAD:
|
|
SMOOTH_L1_BETA: 1.0
|
|
POOLER_SAMPLING_RATIO: 2
|
|
POOLER_TYPE: "ROIAlign"
|
|
INPUT:
|
|
# no scale augmentation
|
|
MIN_SIZE_TRAIN: (800, )
|