update

README.md

@@ -24,6 +24,11 @@
 * 视频示例：   
 ![videoHat](https://codechina.csdn.net/EricLee/yolo_v3/-/raw/master/samples/hat.gif)      
 
+### 5、交通工具检测
+包括：bicycle，car，motorcycle，airplane，bus，train，truck，boat     
+安全帽检测示例如下 ：      
+* 视频示例：  
+![auto](https://codechina.csdn.net/EricLee/yolo_v3/-/raw/master/samples/auto.gif)    
 
 ## 项目配置  
 * 作者开发环境：  
@@ -59,6 +64,21 @@ Year = {2016}}
 该项目采用的是数据集 VOC2028 ，其地址为 https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset     
 * [该项目制作的训练集的数据集下载地址(百度网盘 Password: y694 )](https://pan.baidu.com/s/1m1ysN5r0wYlIUY6FW1XkHg)      
 
+### 5、交通工具数据集  
+该项目采用的是开源数据集 COCO ，其地址为 https://cocodataset.org/   
+* [该项目制作的训练集的数据集下载地址(百度网盘 Password:  )]()    
+  该项目制作数据集类型及对应的实例数量如下所示：  
+```
+  bicycle : 4955
+  car : 30785
+  motorcycle : 6021
+  airplane : 3833
+  bus : 4327
+  train : 3159
+  truck : 7050
+  boat : 7590
+```
+
 ### 数据格式  
 size是全图分辨率， (x，y) 是目标物体中心对于全图的归一化坐标，w,h是目标物体边界框对于全图的归一化宽、高。   
 

show_yolo_anno.py

@@ -18,8 +18,11 @@ if __name__ == "__main__":
     # path='./yolo_person_train/anno/train.txt'
     # path_voc_names = './cfg/person.names'
 
-    path='./yolo_helmet_train/anno/train.txt'
-    path_voc_names = './cfg/helmet.names'
+    # path='./yolo_helmet_train/anno/train.txt'
+    # path_voc_names = './cfg/helmet.names'
+
+    path='./yolo_transport_train/anno/train.txt'
+    path_voc_names = './cfg/transport.names'
 
     with open(path_voc_names, 'r') as f:
         label_map = f.readlines()