diff --git a/README.md b/README.md index 9d3a96d801883a4f54208dfb6b04c7645bb59392..5231b7e147c801d118ef89adb8eb3340a919a2db 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,5 @@ # YOLO V3 -物体检测,包括手部检测、人脸检测、人检测,因为数据集的独立所以分别为3个独立模型。 +物体检测,包括手部检测、人脸检测、人检测、安全帽检测,因为数据集的独立所以分别为4个独立模型。 ## 项目介绍 ### 1、手部检测 @@ -19,6 +19,11 @@ * 视频示例: ![videoPerson](https://codechina.csdn.net/EricLee/yolo_v3/-/raw/master/samples/person.gif) +### 4、安全帽检测 +安全帽检测示例如下 : +* 视频示例: ![videoHat](https://codechina.csdn.net/EricLee/yolo_v3/-/raw/master/samples/hat.gif) + + ## 项目配置 * 作者开发环境: * Python 3.7 @@ -49,6 +54,10 @@ Year = {2016}} 该项目采用的是开源数据集 COCO ,其地址为 https://cocodataset.org/ * [该项目制作的训练集的数据集下载地址(百度网盘 Password: ilv9 )](https://pan.baidu.com/s/1Y1LA8Coc902zrDgmVqAU3A) +### 4、安全帽检测数据集 +该项目采用的是数据集 VOC2028 ,其地址为 https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset +* [该项目制作的训练集的数据集下载地址(百度网盘 Password: y694 )](https://pan.baidu.com/s/1m1ysN5r0wYlIUY6FW1XkHg) + ### 数据格式 size是全图分辨率, (x,y) 是目标物体中心对于全图的归一化坐标,w,h是目标物体边界框对于全图的归一化宽、高。 @@ -78,6 +87,8 @@ h = h*dh ### 3、人检测预训练模型 * [预训练模型下载地址(百度网盘 Password: ise9 )](https://pan.baidu.com/s/1mxiI-tOpE3sU-9TVPJmPWw) +### 4、安全帽检测预训练模型 +* [预训练模型下载地址(百度网盘 Password: )]() ## 项目使用方法 diff --git a/samples/hat.gif b/samples/hat.gif new file mode 100644 index 0000000000000000000000000000000000000000..a6d3838e65e67db4ec6c3b6c292680ac4b762026 Binary files /dev/null and b/samples/hat.gif differ diff --git a/show_yolo_anno.py b/show_yolo_anno.py index 6f4a13be740855dfaf404bc0042297a80d9917f9..5bcb70701c33b9d31c49162ae5bb22a6e40c2bcd 100644 --- a/show_yolo_anno.py +++ b/show_yolo_anno.py @@ -15,8 +15,11 @@ if __name__ == "__main__": # path='./yolo_widerface_open_train/anno/train.txt' # path_voc_names = './cfg/face.names' - path='./yolo_person_train/anno/train.txt' - path_voc_names = './cfg/person.names' + # 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' with open(path_voc_names, 'r') as f: label_map = f.readlines()