Skip to content

Commit

Permalink
add dynamic input for onnx part compatiability.
Browse files Browse the repository at this point in the history
  • Loading branch information
Chenxiaon Niu committed Oct 15, 2020
1 parent e6067ed commit 309c1de
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 0 deletions.
27 changes: 27 additions & 0 deletions configs/COCO-Detection/retinanet_R_50_FPN_1x_bulb.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
_BASE_: "../Base-RetinaNet.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
RETINANET:
NUM_CLASSES: 18
DATASETS:
TRAIN: ("bulb_train",)
TEST: ("bulb_val",)
INPUT:
FORMAT: "RGB"
DYNAMIC: true
# MAX_SIZE_TRAIN: 1920
# MIN_SIZE_TRAIN: (672, 736, 800, 864, 928, 992, 1056)
# MAX_SIZE_TEST: 1920
# MIN_SIZE_TEST: 1056
SOLVER:
IMS_PER_BATCH: 24
LR_SCHEDULER_NAME: "WarmupCosineLR"
BASE_LR: 2.5e-4
MAX_ITER: 364000
CHECKPOINT_PERIOD: 3640
TEST:
EVAL_PERIOD: 1820
OUTPUT_DIR: "./output/retinanet_r50_bot"
VERSION: 2
1 change: 1 addition & 0 deletions detectron2/config/defaults.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,7 @@
# The ground truth mask format that the model will use.
# Mask R-CNN supports either "polygon" or "bitmask" as ground truth.
_C.INPUT.MASK_FORMAT = "polygon" # alternative: "bitmask"
_C.INPUT.DYNAMIC = False


# -----------------------------------------------------------------------------
Expand Down

0 comments on commit 309c1de

Please sign in to comment.