Skip to content

Commit

Permalink
Add FPN for VOC
Browse files Browse the repository at this point in the history
Reviewed By: rbgirshick

Differential Revision: D18046599

fbshipit-source-id: 45829ea7416a845d08865a7303522b74675a7bb3
  • Loading branch information
ppwwyyxx authored and facebook-github-bot committed Oct 21, 2019
1 parent 8ef429d commit 8863132
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions configs/PascalVOC-Detection/faster_rcnn_R_50_FPN.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
_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

0 comments on commit 8863132

Please sign in to comment.