From b33838c0f56489d33a35348217925180c2b79d36 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Mon, 19 Jul 2021 10:42:25 +0800
Subject: [PATCH] add yolov3 original model (#3643)

---
 .../yolov3_darknet53_original_270e_coco.yml   | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 configs/yolov3/yolov3_darknet53_original_270e_coco.yml

diff --git a/configs/yolov3/yolov3_darknet53_original_270e_coco.yml b/configs/yolov3/yolov3_darknet53_original_270e_coco.yml
new file mode 100644
index 000000000..f245d3b1c
--- /dev/null
+++ b/configs/yolov3/yolov3_darknet53_original_270e_coco.yml
@@ -0,0 +1,40 @@
+_BASE_: [
+  '../datasets/coco_detection.yml',
+  '../runtime.yml',
+  '_base_/optimizer_270e.yml',
+  '_base_/yolov3_darknet53.yml',
+  '_base_/yolov3_reader.yml',
+]
+
+snapshot_epoch: 5
+weights: output/yolov3_darknet53_270e_coco/model_final
+
+norm_type: bn
+
+YOLOv3Loss:
+  ignore_thresh: 0.5
+  downsample: [32, 16, 8]
+  label_smooth: false
+
+TrainReader:
+  inputs_def:
+    num_max_boxes: 50
+  sample_transforms:
+    - Decode: {}
+    - RandomDistort: {}
+    - RandomExpand: {fill_value: [123.675, 116.28, 103.53], ratio: 2.0}
+    - RandomCrop: {}
+    - RandomFlip: {}
+  batch_transforms:
+    - BatchRandomResize: {target_size: [320, 352, 384, 416, 448, 480, 512, 544, 576, 608], random_size: True, random_interp: True, keep_ratio: False}
+    - NormalizeBox: {}
+    - PadBox: {num_max_boxes: 50}
+    - BboxXYXY2XYWH: {}
+    - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
+    - Permute: {}
+    - Gt2YoloTarget: {anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]], anchors: [[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326]], downsample_ratios: [32, 16, 8], iou_thresh: 0.5}
+  batch_size: 8
+  shuffle: true
+  drop_last: true
+  mixup_epoch: -1
+  use_shared_memory: true
-- 
GitLab