完善人、头盔、头的权重，使用 yolov5 测试出人体，在进行对人、头盔、头的训练

PeterH0323 · Aug 1, 2020 · a5c6293 · a5c6293
1 parent 520717a
commit a5c6293
Show file tree

Hide file tree

Showing 15 changed files with 20 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,10 +1,10 @@
 # Smart_Construction
 该项目是使用 YOLOv5 来训练在智能工地安全领域中头盔目标检测的应用
 
-![](https://github.com/PeterH0323/Smart_Construction/blob/master/doc/output_1.png?raw=true)
+![](https://github.com/PeterH0323/Smart_Construction/blob/master/doc/output_1.jpg?raw=true)
 ![](https://github.com/PeterH0323/Smart_Construction/blob/master/doc/output_2.jpg?raw=true)
 ![](https://github.com/PeterH0323/Smart_Construction/blob/master/doc/output_3.jpg?raw=true)
-![](https://github.com/PeterH0323/Smart_Construction/blob/master/doc/output_4.png?raw=true)
+![](https://github.com/PeterH0323/Smart_Construction/blob/master/doc/output_4.jpg?raw=true)
 
 # YOLO v5训练自己数据集教程
 使用的数据集：[Safety-Helmet-Wearing-Dataset](https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset) ，感谢这位大神的开源数据集！
@@ -123,3 +123,18 @@ $ python detect.py --source 0  # webcam
                             rtsp://170.93.143.139/rtplive/470011e600ef003a004ee33696235daa  # rtsp stream
                             http://112.50.243.8/PLTV/88888888/224/3221225900/1.m3u8  # http stream
 ```
+
+
+## 检测危险区域内是否有人
+
+### 危险区域标注方式
+我这里使用的是 [精灵标注助手](http://www.jinglingbiaozhu.com/) 标注，生成了对应图片的 json 文件
+
+### 执行侦测，在危险区域里面的任务会被框选出来
+
+侦测图片会保存在 `./inferenct/output/` 文件夹下
+
+运行命令：
+```shell script
+$ python area_detect.py --source ./area_dangerous --weights ./weights/pro_helmet_head_person.pt
+```
diff --git a/area_dangerous/1.jpg b/area_dangerous/1.jpg
diff --git a/area_dangerous/2.jpg b/area_dangerous/2.jpg
diff --git a/area_dangerous/area_labels/1.json b/area_dangerous/area_labels/1.json
@@ -0,0 +1 @@
+{"path":"E:\\AI_Project\\Smart_Construction_Project\\area_dangerous\\1.jpg","outputs":{"object":[{"name":"dangerous","bndbox":{"xmin":484,"ymin":111,"xmax":620,"ymax":290}}]},"time_labeled":1596282882403,"labeled":true,"size":{"width":1023,"height":682,"depth":3}}
diff --git a/area_dangerous/area_labels/2.json b/area_dangerous/area_labels/2.json
@@ -0,0 +1 @@
+{"path":"E:\\AI_Project\\Smart_Construction_Project\\area_dangerous\\2.jpg","outputs":{"object":[{"name":"dangerous","bndbox":{"xmin":265,"ymin":3,"xmax":630,"ymax":418}}]},"time_labeled":1596282897538,"labeled":true,"size":{"width":640,"height":427,"depth":3}}
diff --git a/area_detect.py b/area_detect.py
@@ -100,7 +100,7 @@ def detect(save_img=False):
                     if save_img or view_img:  # Add bbox to image
                         label = '%s %.2f' % (names[int(cls)], conf)
                         if names[int(cls)] == "person":
-                            判断是否在危险区域内
+                            # 判断是否在危险区域内
                             if person_in_area_dangerous(xyxy, Path(p).name) == 1:
                                 # 返回 1 表明是在危险区域，框住人
                                 plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

diff --git a/doc/output_1.jpg b/doc/output_1.jpg
diff --git a/doc/output_1.png b/doc/output_1.png
diff --git a/doc/output_2.jpg b/doc/output_2.jpg
diff --git a/doc/output_3.jpg b/doc/output_3.jpg
diff --git a/doc/output_4.jpg b/doc/output_4.jpg
diff --git a/doc/output_4.png b/doc/output_4.png
diff --git a/doc/test_batch0_gt.jpg b/doc/test_batch0_gt.jpg
diff --git a/weights/helmet_head_person.pt b/weights/helmet_head_person.pt
diff --git a/weights/pro_helmet_head_person.pt b/weights/pro_helmet_head_person.pt
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{"path":"E:\\AI_Project\\Smart_Construction_Project\\area_dangerous\\1.jpg","outputs":{"object":[{"name":"dangerous","bndbox":{"xmin":484,"ymin":111,"xmax":620,"ymax":290}}]},"time_labeled":1596282882403,"labeled":true,"size":{"width":1023,"height":682,"depth":3}}