test_simple.yaml

_BASE_: "../R_50/Base_det.yaml"

MODEL:
  WEIGHTS: "pretrained_models/CTW1500/finetune_ctw_96voc.pth" # 可替换为自己的预训练模型地址
  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: 1.0
        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: 5e-5
  LR_BACKBONE: 5e-6
  WARMUP_ITERS: 0
  STEPS: (8000,)
  MAX_ITER: 12000
  CHECKPOINT_PERIOD: 4000

TEST:
  EVAL_PERIOD: 1000

OUTPUT_DIR: "output/R50/simple/pretrain_maxlen50_96voc"