From db2ddaef4d4227c33814c34474b7898f26f16722 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Wed, 3 Mar 2021 21:52:00 +0800
Subject: [PATCH] add transport cfg

---
 cfg/transport.data  | 15 +++++++++++++++
 cfg/transport.names |  8 ++++++++
 2 files changed, 23 insertions(+)
 create mode 100644 cfg/transport.data
 create mode 100644 cfg/transport.names

diff --git a/cfg/transport.data b/cfg/transport.data
new file mode 100644
index 0000000..8482ece
--- /dev/null
+++ b/cfg/transport.data
@@ -0,0 +1,15 @@
+cfg_model=yolo
+classes=8
+gpus = 0
+num_workers = 8
+batch_size = 8
+img_size = 416
+multi_scale = True
+epochs = 100
+train=./yolo_transport_train/anno/train.txt
+valid=./yolo_transport_train/anno/train.txt
+names=./cfg/transport.names
+#finetune_model=./coco_model/yolov3_coco.pt
+finetune_model = ./weights-yolov3-transport/latest_416.pt
+lr_step = 10,20,30
+lr0 = 0.001
diff --git a/cfg/transport.names b/cfg/transport.names
new file mode 100644
index 0000000..a1f5baf
--- /dev/null
+++ b/cfg/transport.names
@@ -0,0 +1,8 @@
+bicycle
+car
+motorcycle
+airplane
+bus
+train
+truck
+boat
-- 
GitLab