未验证 提交 c07bf620 编写于 作者: F FlyingQianMM 提交者: GitHub

refine pedestrian_yolov3_darknet.yml and vehicle_yolov3_darknet.yml in contrib (#324)

上级 a764cd1e
architecture: YOLOv3 architecture: YOLOv3
train_feed: YoloTrainFeed
eval_feed: YoloEvalFeed
test_feed: YoloTestFeed
use_gpu: true use_gpu: true
max_iters: 200000 max_iters: 200000
log_smooth_window: 20 log_smooth_window: 20
...@@ -11,6 +8,7 @@ metric: COCO ...@@ -11,6 +8,7 @@ metric: COCO
pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/DarkNet53_pretrained.tar pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/DarkNet53_pretrained.tar
weights: https://paddlemodels.bj.bcebos.com/object_detection/pedestrian_yolov3_darknet.tar weights: https://paddlemodels.bj.bcebos.com/object_detection/pedestrian_yolov3_darknet.tar
num_classes: 1 num_classes: 1
use_fine_grained_loss: false
YOLOv3: YOLOv3:
backbone: DarkNet backbone: DarkNet
...@@ -27,8 +25,7 @@ YOLOv3Head: ...@@ -27,8 +25,7 @@ YOLOv3Head:
[30, 61], [62, 45], [59, 119], [30, 61], [62, 45], [59, 119],
[116, 90], [156, 198], [373, 326]] [116, 90], [156, 198], [373, 326]]
norm_decay: 0. norm_decay: 0.
ignore_thresh: 0.7 yolo_loss: YOLOv3Loss
label_smooth: true
nms: nms:
background_label: -1 background_label: -1
keep_top_k: 100 keep_top_k: 100
...@@ -37,6 +34,11 @@ YOLOv3Head: ...@@ -37,6 +34,11 @@ YOLOv3Head:
normalized: false normalized: false
score_threshold: 0.01 score_threshold: 0.01
YOLOv3Loss:
batch_size: 8
ignore_thresh: 0.7
label_smooth: false
LearningRate: LearningRate:
base_lr: 0.001 base_lr: 0.001
schedulers: schedulers:
...@@ -57,26 +59,28 @@ OptimizerBuilder: ...@@ -57,26 +59,28 @@ OptimizerBuilder:
factor: 0.0005 factor: 0.0005
type: L2 type: L2
YoloTrainFeed: _READER_: '../../configs/yolov3_reader.yml'
TrainReader:
batch_size: 8 batch_size: 8
dataset: dataset:
!COCODataSet
dataset_dir: dataset/pedestrian dataset_dir: dataset/pedestrian
annotation: annotations/instances_train2017.json anno_path: annotations/instances_train2017.json
image_dir: train2017 image_dir: train2017
num_workers: 8 with_background: false
bufsize: 128
use_process: true
YoloEvalFeed: EvalReader:
batch_size: 8 batch_size: 8
image_shape: [3, 608, 608]
dataset: dataset:
!COCODataSet
dataset_dir: dataset/pedestrian dataset_dir: dataset/pedestrian
annotation: annotations/instances_val2017.json anno_path: annotations/instances_val2017.json
image_dir: val2017 image_dir: val2017
with_background: false
YoloTestFeed: TestReader:
batch_size: 1 batch_size: 1
image_shape: [3, 608, 608]
dataset: dataset:
annotation: contrib/PedestrianDetection/pedestrian.json !ImageFolder
anno_path: contrib/PedestrianDetection/pedestrian.json
with_background: false
architecture: YOLOv3 architecture: YOLOv3
train_feed: YoloTrainFeed
eval_feed: YoloEvalFeed
test_feed: YoloTestFeed
use_gpu: true use_gpu: true
max_iters: 120000 max_iters: 120000
log_smooth_window: 20 log_smooth_window: 20
...@@ -27,8 +24,7 @@ YOLOv3Head: ...@@ -27,8 +24,7 @@ YOLOv3Head:
[23, 33], [40, 25], [54, 50], [23, 33], [40, 25], [54, 50],
[101, 80], [139, 145], [253, 224]] [101, 80], [139, 145], [253, 224]]
norm_decay: 0. norm_decay: 0.
ignore_thresh: 0.7 yolo_loss: YOLOv3Loss
label_smooth: false
nms: nms:
background_label: -1 background_label: -1
keep_top_k: 100 keep_top_k: 100
...@@ -37,6 +33,11 @@ YOLOv3Head: ...@@ -37,6 +33,11 @@ YOLOv3Head:
normalized: false normalized: false
score_threshold: 0.005 score_threshold: 0.005
YOLOv3Loss:
batch_size: 8
ignore_thresh: 0.7
label_smooth: false
LearningRate: LearningRate:
base_lr: 0.001 base_lr: 0.001
schedulers: schedulers:
...@@ -57,26 +58,28 @@ OptimizerBuilder: ...@@ -57,26 +58,28 @@ OptimizerBuilder:
factor: 0.0005 factor: 0.0005
type: L2 type: L2
YoloTrainFeed: _READER_: '../../configs/yolov3_reader.yml'
TrainReader:
batch_size: 8 batch_size: 8
dataset: dataset:
!COCODataSet
dataset_dir: dataset/vehicle dataset_dir: dataset/vehicle
annotation: annotations/instances_train2017.json anno_path: annotations/instances_train2017.json
image_dir: train2017 image_dir: train2017
num_workers: 8 with_background: false
bufsize: 128
use_process: true
YoloEvalFeed: EvalReader:
batch_size: 8 batch_size: 8
image_shape: [3, 608, 608]
dataset: dataset:
!COCODataSet
dataset_dir: dataset/vehicle dataset_dir: dataset/vehicle
annotation: annotations/instances_val2017.json anno_path: annotations/instances_val2017.json
image_dir: val2017 image_dir: val2017
with_background: false
YoloTestFeed: TestReader:
batch_size: 1 batch_size: 1
image_shape: [3, 608, 608]
dataset: dataset:
annotation: contrib/VehicleDetection/vehicle.json !ImageFolder
anno_path: contrib/VehicleDetection/vehicle.json
with_background: false
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册