diff --git a/README.md b/README.md index caaa2bf254dedd5be7b7b77df5c0abb5afeeb92e..bdc8c1b20bbe24fdc8b0e18da27f3f8838bf80f9 100644 --- a/README.md +++ b/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是目标物体边界框对于全图的归一化宽、高。 diff --git a/show_yolo_anno.py b/show_yolo_anno.py index 5bcb70701c33b9d31c49162ae5bb22a6e40c2bcd..acec73c54f48a98370654bfc39c8e0bb27b197ed 100644 --- a/show_yolo_anno.py +++ b/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()