ssd_mobilenet_v1_voc.yml 1.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
architecture: SSD
max_iters: 28000
train_feed: SSDTrainFeed
eval_feed: SSDEvalFeed
test_feed: SSDTestFeed
pretrain_weights: ./ssd3/

use_gpu: true
snapshot_iter: 2000
log_smooth_window: 1
metric: VOC
save_dir: output
weights: output/ssd_mobilenet_v1_voc/model_final/

SSD:
  backbone: MobileNet
  multi_box_head: MultiBoxHead
  num_classes: 21
19
  metric:
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
    ap_version: 11point
    evaluate_difficult: false
    overlap_threshold: 0.5
  output_decoder:
    background_label: 0
    keep_top_k: 200
    nms_eta: 1.0
    nms_threshold: 0.45
    nms_top_k: 400
    score_threshold: 0.01

MobileNet:
  norm_decay: 0.
  conv_group_scale: 1
  extra_block_filters: [[256, 512], [128, 256], [128, 256], [64, 128]]
  with_extra_blocks: true

MultiBoxHead:
  aspect_ratios: [[2.], [2., 3.], [2., 3.], [2., 3.], [2., 3.], [2., 3.]]
  base_size: 300
  flip: true
  max_ratio: 90
  max_sizes: [[], 150.0, 195.0, 240.0, 285.0, 300.0]
  min_ratio: 20
  min_sizes: [60.0, 105.0, 150.0, 195.0, 240.0, 285.0]
  offset: 0.5

LearningRate:
  schedulers:
  - !PiecewiseDecay
    milestones: [10000, 15000, 20000, 25000]
    values: [0.001, 0.0005, 0.00025, 0.0001, 0.00001]

OptimizerBuilder:
  optimizer:
    momentum: 0.0
    type: RMSPropOptimizer
  regularizer:
    factor: 0.00005
    type: L2
60

61 62 63 64
SSDTrainFeed:
  batch_size: 32
  use_process: true
  dataset:
65
    dataset_dir: dataset/voc
66 67
    annotation: VOCdevkit/VOC_all/ImageSets/Main/train.txt
    image_dir: VOCdevkit/VOC_all/JPEGImages
K
Kaipeng Deng 已提交
68
    use_default_label: true
69 70 71 72 73

SSDEvalFeed:
  batch_size: 64
  use_process: true
  dataset:
74
    dataset_dir: dataset/voc
75 76
    annotation: VOCdevkit/VOC_all/ImageSets/Main/val.txt
    image_dir: VOCdevkit/VOC_all/JPEGImages
K
Kaipeng Deng 已提交
77
    use_default_label: true
78 79 80 81 82
  drop_last: false

SSDTestFeed:
  batch_size: 1
  dataset:
K
Kaipeng Deng 已提交
83
    use_default_label: true
84
  drop_last: false