mask_rcnn_r50_1x.yml 387 字节
Newer Older
1 2 3 4 5 6 7 8
_BASE_: "configs/base/mask_rcnn_r50_1x.yml"

use_gpu: true
worker_num: 0
log_smooth_window: 20
save_dir: output
metric: COCO
num_classes: 81
S
sunxl1988 已提交
9
log_iter: 20
10

S
sunxl1988 已提交
11 12
epoch: 12

13
TrainReader:
S
sunxl1988 已提交
14
  batch_size: 2
S
sunxl1988 已提交
15
  shuffle: true
16 17 18 19
  inputs_def:
    fields: ['image', 'im_info', 'im_id', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly']
  dataset:
    name: COCODataset
S
sunxl1988 已提交
20
    dataset_dir: /home/ai/dataset/COCO17