From f59aa272e30443701561ae787dce4226b358b6a2 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Mon, 15 Mar 2021 11:19:31 +0800
Subject: [PATCH] add helmet cfg files

---
 cfg/helmet.data  | 15 +++++++++++++++
 cfg/helmet.names |  2 ++
 2 files changed, 17 insertions(+)
 create mode 100644 cfg/helmet.data
 create mode 100644 cfg/helmet.names

diff --git a/cfg/helmet.data b/cfg/helmet.data
new file mode 100644
index 0000000..47ec576
--- /dev/null
+++ b/cfg/helmet.data
@@ -0,0 +1,15 @@
+cfg_model=yolo
+classes=2
+gpus = 0
+num_workers = 8
+batch_size = 8
+img_size = 416
+multi_scale = True
+epochs = 100
+train=./yolo_helmet_train/anno/train.txt
+valid=./yolo_helmet_train/anno/train.txt
+names=./cfg/helmet.names
+#finetune_model=./coco_model/yolov3_coco.pt
+finetune_model = ./weights-yolov3-helmet/latest_416.pt
+lr_step = 10,20,30
+lr0 = 0.0001
diff --git a/cfg/helmet.names b/cfg/helmet.names
new file mode 100644
index 0000000..18aeb08
--- /dev/null
+++ b/cfg/helmet.names
@@ -0,0 +1,2 @@
+face
+hat
-- 
GitLab