diff --git a/configs/base/mask_rcnn_r50_1x.yml b/configs/base/mask_rcnn_r50_1x.yml index f8a67965b741ac9677c3e9da3f0da5e7d5e45cc0..17fc0c0c3f267d75bcc1d05c2a6c45a9b19f39f2 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 2559664621fef23667ad404045662b52dc14d0a7..dfc499c1316249885d0983e7f911ca5651a3625f 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 1a248a8261f778493ad322da103f378218b1f172..b10d5db08bc5fe7bfb1420aa6fde662a64a61a5d 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 e12ed9a634c51e6cbadc1690e062382ff81c1ad0..1664b156b32ff8e9fd93f849b305939f063db955 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