test.yml

MODEL:
  SSDS: YOLOV3
  NETS: ResNet18
  IMAGE_SIZE: [320, 320]
  NUM_CLASSES: 80
  FEATURE_LAYER: [[3, 4, 5], [128, 256, 512]]
  SIZES: [[2.0, 2.828], [2.0, 2.828], [2.0, 4.0, 8.0]]
  ASPECT_RATIOS: [[1, 2, 0.5], [1, 2, 0.5], [1, 2, 0.5]]

TRAIN:
  MAX_EPOCHS: 50
  CHECKPOINTS_EPOCHS: 1
  BATCH_SIZE: 1
  TRAINABLE_SCOPE: 'transforms,extras,loc,conf'
  RESUME_SCOPE: ''
  OPTIMIZER:
    OPTIMIZER: adam
    LEARNING_RATE: 0.0001
    MOMENTUM: 0.9
    WEIGHT_DECAY: 0.0001
  LR_SCHEDULER:
    SCHEDULER: exponential
    GAMMA: 0.97
    WARM_UP_EPOCHS: 0

TEST:
  BATCH_SIZE: 1
  TEST_SCOPE: [49, 50]

MATCHER:
  LOCATE_LOSS: "CIOULoss"

POST_PROCESS:
  SCORE_THRESHOLD: 0.01
  IOU_THRESHOLD: 0.6
  MAX_DETECTIONS: 100

DATASET:
  DATASET: 'DaliCOCO'
  DATASET_DIR: '/data1/dataset/tiny_coco'
  TRAIN_SETS: ["train2017"]
  TEST_SETS:  ["val2017"]
  DEVICE_ID: [0]

EXP_DIR: './experiments/models/pretrain/yolo_resnet18_coco_test2'
LOG_DIR: './experiments/models/pretrain/yolo_resnet18_coco_test2'
# RESUME_CHECKPOINT: './experiments/weights/pretrain/SSD_ResNet18_300x300_COCO_20.4.pth'
PHASE: ['train']
DEVICE_ID: [0]