From 9ba7b898ececa611a9893ba0dd1627db7fa08bfa Mon Sep 17 00:00:00 2001 From: sunxl1988 <935448535@qq.com> Date: Thu, 20 Aug 2020 07:46:40 +0000 Subject: [PATCH] test=dygraph update cfg --- configs/base/mask_rcnn_r50_1x.yml | 1 + configs/base/mask_rcnn_r50_fpn_1x.yml | 3 ++- configs/base/yolov3_darknet.yml | 4 ++-- configs/example/mask_rcnn_r50_1x.yml | 4 ++-- 4 files changed, 7 insertions(+), 5 deletions(-) diff --git a/configs/base/mask_rcnn_r50_1x.yml b/configs/base/mask_rcnn_r50_1x.yml index f8a67965b..17fc0c0c3 100644 --- a/configs/base/mask_rcnn_r50_1x.yml +++ b/configs/base/mask_rcnn_r50_1x.yml @@ -6,6 +6,7 @@ epoch: 12 use_prefetch: False worker_num: 0 log_smooth_window: 20 +log_iter: 20 save_dir: output metric: COCO num_classes: 81 diff --git a/configs/base/mask_rcnn_r50_fpn_1x.yml b/configs/base/mask_rcnn_r50_fpn_1x.yml index 255966462..dfc499c13 100644 --- a/configs/base/mask_rcnn_r50_fpn_1x.yml +++ b/configs/base/mask_rcnn_r50_fpn_1x.yml @@ -6,10 +6,11 @@ epoch: 12 use_prefetch: False worker_num: 0 log_smooth_window: 20 +log_iter: 20 save_dir: output metric: COCO num_classes: 81 -load_static_weights: false +load_static_weights: true _READER_: '../reader/mask_rcnn.yml' _ARCHITECHTURE_: '../architechture/mask_rcnn_fpn.yml' diff --git a/configs/base/yolov3_darknet.yml b/configs/base/yolov3_darknet.yml index 1a248a826..b10d5db08 100644 --- a/configs/base/yolov3_darknet.yml +++ b/configs/base/yolov3_darknet.yml @@ -6,12 +6,12 @@ worker_num: 0 use_prefetch: False epoch: 270 log_smooth_window: 20 +log_iter: 20 save_dir: output metric: COCO num_classes: 80 use_fine_grained_loss: false -open_debug: False - +load_static_weights: true _READER_: '../reader/yolo.yml' _ARCHITECHTURE_: '../architechture/yolov3.yml' diff --git a/configs/example/mask_rcnn_r50_1x.yml b/configs/example/mask_rcnn_r50_1x.yml index e12ed9a63..1664b156b 100644 --- a/configs/example/mask_rcnn_r50_1x.yml +++ b/configs/example/mask_rcnn_r50_1x.yml @@ -11,10 +11,10 @@ log_iter: 20 epoch: 12 TrainReader: - batch_size: 1 + batch_size: 2 shuffle: true inputs_def: fields: ['image', 'im_info', 'im_id', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_poly'] dataset: name: COCODataset - dataset_dir: /dataset/coco + dataset_dir: /home/ai/dataset/COCO17 -- GitLab