From feef394dc59688f28dfa44d05338a69e532e91c4 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Mon, 16 Mar 2020 10:39:47 +0800 Subject: [PATCH] fix yolo configs (#332) --- configs/yolov3_darknet.yml | 2 +- configs/yolov3_darknet_voc.yml | 2 +- configs/yolov3_mobilenet_v1.yml | 2 +- configs/yolov3_mobilenet_v1_voc.yml | 1 - configs/yolov3_r34.yml | 2 +- configs/yolov3_r34_voc.yml | 1 - 6 files changed, 4 insertions(+), 6 deletions(-) diff --git a/configs/yolov3_darknet.yml b/configs/yolov3_darknet.yml index 22ae6d914..2ea5ba7bd 100644 --- a/configs/yolov3_darknet.yml +++ b/configs/yolov3_darknet.yml @@ -37,7 +37,7 @@ YOLOv3Head: YOLOv3Loss: batch_size: 8 ignore_thresh: 0.7 - label_smooth: false + label_smooth: true LearningRate: base_lr: 0.001 diff --git a/configs/yolov3_darknet_voc.yml b/configs/yolov3_darknet_voc.yml index f8ff843bb..241d12934 100644 --- a/configs/yolov3_darknet_voc.yml +++ b/configs/yolov3_darknet_voc.yml @@ -38,7 +38,7 @@ YOLOv3Head: YOLOv3Loss: batch_size: 8 ignore_thresh: 0.7 - label_smooth: true + label_smooth: false LearningRate: base_lr: 0.001 diff --git a/configs/yolov3_mobilenet_v1.yml b/configs/yolov3_mobilenet_v1.yml index e27900b8b..a66b26c72 100644 --- a/configs/yolov3_mobilenet_v1.yml +++ b/configs/yolov3_mobilenet_v1.yml @@ -38,7 +38,7 @@ YOLOv3Head: YOLOv3Loss: batch_size: 8 ignore_thresh: 0.7 - label_smooth: false + label_smooth: true LearningRate: base_lr: 0.001 diff --git a/configs/yolov3_mobilenet_v1_voc.yml b/configs/yolov3_mobilenet_v1_voc.yml index 44cc9ed6f..2ce5e40bf 100644 --- a/configs/yolov3_mobilenet_v1_voc.yml +++ b/configs/yolov3_mobilenet_v1_voc.yml @@ -72,7 +72,6 @@ TrainReader: EvalReader: inputs_def: - image_shape: [3, 608, 608] fields: ['image', 'im_size', 'im_id', 'gt_bbox', 'gt_class', 'is_difficult'] num_max_boxes: 50 dataset: diff --git a/configs/yolov3_r34.yml b/configs/yolov3_r34.yml index 765190802..4cc05f24d 100644 --- a/configs/yolov3_r34.yml +++ b/configs/yolov3_r34.yml @@ -40,7 +40,7 @@ YOLOv3Head: YOLOv3Loss: batch_size: 8 ignore_thresh: 0.7 - label_smooth: false + label_smooth: true LearningRate: base_lr: 0.001 diff --git a/configs/yolov3_r34_voc.yml b/configs/yolov3_r34_voc.yml index 8022ffd16..f1cb5a4a2 100644 --- a/configs/yolov3_r34_voc.yml +++ b/configs/yolov3_r34_voc.yml @@ -74,7 +74,6 @@ TrainReader: EvalReader: inputs_def: - image_shape: [3, 608, 608] fields: ['image', 'im_size', 'im_id', 'gt_bbox', 'gt_class', 'is_difficult'] num_max_boxes: 50 dataset: -- GitLab