Merge pull request #6 from dongxuecheng/dxc/basic_frame

增加权重文件，修改部分代码

README.md

@@ -1,9 +1,9 @@
 # self_rescuer
-自救器
-# 测试-dxc-9-2
-测试-dxc-9-2
-# 测试协同
+## 压缩氧
 
-hello lxd
-
-hello lxd2
+1 = 外壳去掉
+2 = 脖带戴好 
+3 = 咬口具
+4 = 氧气瓶开启
+5 = 补气
+6 = 带鼻夹

app.py

@@ -1,6 +1,7 @@
 import threading
 import time
 from flask import Flask, jsonify,send_from_directory
+from compressed_oxygen_detect import init_compressed_oxygen_detection,start_compressed_oxygen_detection
 from globals import inference_thread, stop_event,lock,redis_client
 
 #焊接考核的穿戴
@@ -96,4 +97,4 @@ def get_image(filename):
 if __name__ == '__main__':
 
     # Start the Flask server
-    app.run(debug=False, host='172.16.20.163', port=5001)
+    app.run(debug=False, host='172.16.20.163', port=5007)

welding_wearing_detect.py → compressed_oxygen_detect.py

@@ -49,7 +49,8 @@ def process_video(model_path, video_source, start_event):
                     confidence = confidences[i].item()
                     cls = int(classes[i].item())
                     label = model.names[cls]
-                pass
+                    print("label:",label)
+                #pass
                 
                     
                 start_event.set()    

config.py

@@ -8,12 +8,12 @@ SAVE_IMG_PATH = '/mnt/xcd/code/ai_test/static/images'  # 图片保存在服务
 POST_IMG_PATH1 = 'http://172.16.20.163:5001/images'  # 通过端口映射post发送能够访问的位置 焊接考核科目1
 POST_IMG_PATH2 = 'http://172.16.20.163:5002/images' #焊接考核科目2
 
-VIDEO_SOURCE = '/mnt/xcd/code/ai_test/static/videos/1.mp4'  # 视频源路径
+VIDEO_SOURCE = 'rtsp://admin:yaoan1234@172.16.22.233/cam/realmonitor?channel=1&subtype=0'  # 视频源路径
 #焊接考核视频流
 # Define paths to RTSP streams
 
 
-MODEL_PATH="/mnt/xcd/code/ai_test/weights/ch1_welding_switch_813.pt"
+MODEL_PATH="/mnt/xcd/code/self_rescuer/weights/self_rescuer.pt"
 
 
 

post_demo.py

@@ -0,0 +1,11 @@
+import requests
+
+url = 'http://172.16.20.163:5007/compressed_oxygen_detection'
+
+
+response = requests.get(url)
+
+if response.status_code == 200:
+    print(response.text)
+else:
+    print(response.text)