Commit cce6e1bf authored by chenych's avatar chenych
Browse files

First commit.

parents
Pipeline #640 failed with stages
in 0 seconds
_BASE_: "../R_50/Base_det.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/torchvision/R-50.pkl"
TRANSFORMER:
VOC_SIZE: 96
NUM_POINTS: 50
LOSS:
BEZIER_SAMPLE_POINTS: 50 # the same as NUM_POINTS
BEZIER_CLASS_WEIGHT: 1.0
BEZIER_COORD_WEIGHT: 0.5
POINT_CLASS_WEIGHT: 1.0
POINT_COORD_WEIGHT: 0.5
POINT_TEXT_WEIGHT: 0.5
BOUNDARY_WEIGHT: 0.25
DATASETS:
TRAIN: ("simple_train",)
TEST: ("simple_test",)
INPUT:
MIN_SIZE_TEST: 1000
MAX_SIZE_TEST: 1200
SOLVER:
IMS_PER_BATCH: 8
BASE_LR: 1e-4
LR_BACKBONE: 1e-5
WARMUP_ITERS: 0
STEPS: (320000,)
MAX_ITER: 375000
CHECKPOINT_PERIOD: 100000
TEST:
EVAL_PERIOD: 10000
OUTPUT_DIR: "output/R50/ctw1500/pretrain_150k-tt-mlt-ic13-15_maxlen50_96voc"
This source diff could not be displayed because it is too large. You can view the blob instead.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment