Bank_second_part/xbank_detect_process/.gitignore

@@ -1,2 +0,0 @@
-tensort_cache
-__pycache__

Bank_second_part/xbank_detect_process/.vscode/launch.json

@@ -1,18 +0,0 @@
-{
-    // Use IntelliSense to learn about possible attributes.
-    // Hover to view descriptions of existing attributes.
-    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
-    "version": "0.2.0",
-    "configurations": [
-        {
-            "name": "Python: Current File",
-            "type": "python",
-            "request": "launch",
-            "program": "${file}",
-            "console": "integratedTerminal",
-            "cwd": "${fileDirname}",
-            "justMyCode": true,
-            "purpose": ["debug-in-terminal"],
-        }
-    ]
-}

Bank_second_part/xbank_detect_process/.vscode/settings.json

@@ -1,13 +0,0 @@
-{
-    "cSpell.words": [
-        "fastdeploy",
-        "imread",
-        "imshow",
-        "rtsp",
-        "yolov"
-    ],
-    "[python]": {
-        "editor.defaultFormatter": "ms-python.black-formatter"
-    },
-    "python.formatting.provider": "none"
-}

Bank_second_part/xbank_detect_process/analysis_data/config_load.py

@@ -1,7 +1,7 @@
 import yaml
 from pathlib import Path
 
-def get_configs(yaml_files):
+def get_configs(ymal_files):
-    yaml_path = Path(__file__).parent.parent / yaml_files
+    yaml_path = Path(__file__).parent / ymal_files
     with yaml_path.open("r", encoding="utf-8") as f:
         return yaml.load(f, Loader=yaml.FullLoader)

Bank_second_part/xbank_detect_process/analysis_data/data_rtsp.py

@@ -0,0 +1,9 @@
+
+
+def rtsp_para(scource):
+
+    if scource.split('://')[0] == 'rtsp':
+        return True
+    else:
+        return False
+

Bank_second_part/xbank_detect_process/analysis_result/get_model_result.py

@@ -2,8 +2,6 @@
 
 def det_img(model_inference,images_frame,confidence,label_name_list):
 
-    assert images_frame is not None
-
     result = model_inference.predict(images_frame)
 
     bbox_list = result.boxes

Bank_second_part/xbank_detect_process/config_det.yaml

@@ -1,18 +1,18 @@
 detect_0: 
-  source : /home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi
+  source : rtsp://admin:@192.168.10.18
-  model: ./config_phone.yaml
+  model: /home/yaxin/xbank/xbank_poc_test/config_phone.yaml
 
-# detect_1: 
+detect_1: 
-#   source : rtsp://admin:@192.168.10.12
+  source : rtsp://admin:@192.168.10.12
-#   model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
+  model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
 
-# detect_2: 
+detect_2: 
-#   source : rtsp://admin:@192.168.10.18
+  source : rtsp://admin:@192.168.10.18
-#   model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
+  model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
 
-# detect_3: 
+detect_3: 
-#   source : rtsp://admin:@192.168.10.11
+  source : rtsp://admin:@192.168.10.11
-#   model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
+  model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
 
 # detect_4: 
 #   source : rtsp://admin:@192.168.10.11

Bank_second_part/xbank_detect_process/config_phone.yaml

@@ -1,6 +1,6 @@
 # load model file
-model: ./model_file/yolov5.onnx
+model: /home/yaxin/xbank/xbank_poc_test/model_file/yolov5.onnx
-model_cache: ./tensort_cache/yolov5.trt
+model_cache: /home/yaxin/xbank/xbank_poc_test/tensort_cache/yolov5.trt
 
 # label and bbox message set 
 model_parameter:
@@ -11,7 +11,7 @@ model_parameter:
   relevancy_para : 0
   confidence : 0.2
 
-save_path : ./save_path/hands
+save_path : /home/yaxin/xbank/xbank_poc_test/save_path/hands
 # save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/hands
 # test_path : /home/yaxin/xbank/xbank_poc_test/test_save_path/hands
 save_path_original : False

Bank_second_part/xbank_detect_process/config_sleep.yaml

@@ -1,6 +1,6 @@
 # load model file
-model: ./model_file/yolov8.onnx
+model: /home/yaxin/xbank/xbank_poc_test/model_file/yolov8.onnx
-model_cache: ./tensort_cache/yolov8.trt
+model_cache: /home/yaxin/xbank/xbank_poc_test/tensort_cache/yolov8.trt
 
 # label and bbox message set 
 model_parameter:
@@ -11,7 +11,7 @@ model_parameter:
   relevancy_para : False
   confidence : 0.2
 
-save_path : ./save_path/sleep
+save_path : /home/yaxin/xbank/xbank_poc_test/save_path/sleep
 # save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/sleep
 # test_path : /home/yaxin/xbank/xbank_poc_test/test_save_path/sleep
 save_path_original : False

Bank_second_part/xbank_detect_process/main.py

@@ -1,6 +1,9 @@
 #coding: utf-8
 from main_process import data_load
+import multiprocessing
 from analysis_data.config_load import get_configs
+from model_load.model_load import Load_model
+import time
 from multiprocessing import Pool
 
 
@@ -22,14 +25,12 @@ def get_args_list(args_data):
 if __name__ == "__main__":
 
     # 加载配置文件
-    args = './config_det.yaml'
+    args = '/home/yaxin/xbank/xbank_poc_test/config_det.yaml'
-    args_data = get_configs(yaml_files=args)
+    args_data = get_configs(ymal_files=args)
     args_list = get_args_list(args_data)
 
     process_num = len(args_list)
 
-    print(f"args_list:{args_list}")
-
     with Pool(process_num) as pool:
         pool.map(data_load, args_list)
     

Bank_second_part/xbank_detect_process/main_process.py

@@ -1,13 +1,16 @@
 from analysis_result.get_model_result import det_img
 from analysis_result.same_model_img import same_model_img_analysis_labels, model_labels_selet
-from model_load.model_load import load_model
+from model_load.model_load import Load_model
 from drawing_img.drawing_img import drawing_frame
-from analysis_data.data_dir_file import get_dir_file
+from analysis_data.data_rtsp import rtsp_para
+from analysis_data.data_dir_file import get_dir_file, get_imgframe
+from analysis_data.data_load import data_load
 from analysis_data.config_load import get_configs
-from utils import is_image_file, is_rtsp_or_video
 
+import yaml
 import cv2
 import os
+from pathlib import Path
 import time
 from datetime import datetime
 
@@ -17,20 +20,20 @@ def data_load(args):
     # print('正在运行的进程',msg)
     # print(args)
     source = args[0]
-    model_yaml = args[1]
+    model_ymal = args[1]
 
     # 数据加载
-    rtsp_or_video_source = is_rtsp_or_video(source)
+    rtsp_source = rtsp_para(source)
     dir_source = os.path.isdir(source)
-    img_source = is_image_file(source)
+    file_source = os.path.isfile(source)
 
     # # 模型加载
-    model_data = get_configs(model_yaml)
+    model_data = get_configs(model_ymal)
-    model_inference = load_model(model_file=model_data["model"],
+    model_inference = Load_model(model_file=model_data["model"],
                                  device=model_data["model_parameter"]['device'],
                                  cache_file=model_data["model_cache"])
 
-    if rtsp_or_video_source:
+    if rtsp_source:
 
         cap = cv2.VideoCapture(source)
 
@@ -46,11 +49,15 @@ def data_load(args):
 
                 #     cv2.imshow('18',frame)
 
-                img_frame_dict = {"path": source, 'frame': frame}
+                t1 = time.time()
+                imgframe_dict = {"path": source, 'frame': frame}
                 images_update = img_process(
-                    img_frame_dict, model_inference, model_data)
+                    imgframe_dict, model_inference, model_data)
                 
                 # print(type(images_update['frame']))
+                t2 = time.time()
+                tx = t2 - t1
+                print('检测一张图片的时间为：',tx)
 
                 # if source == 'rtsp://admin:@192.168.10.18':
                 #     cv2.namedWindow('18',0)
@@ -80,10 +87,10 @@ def data_load(args):
                 t1 = time.time()
                 images = cv2.imread(img)   
  
-                img_frame_dict = {"path": img, 'frame': images}
+                imgframe_dict = {"path": img, 'frame': images}
 
                 images_update = img_process(
-                    img_frame_dict, model_inference, model_data)
+                    imgframe_dict, model_inference, model_data)
 
                 t2 = time.time()
                 tx = t2 - t1
@@ -94,22 +101,21 @@ def data_load(args):
 
             pass
 
-    if img_source:
+    if file_source:
 
         img_para = True
 
         if img_para:
             images = cv2.imread(source)
-            img_frame_dict = {"path": source, 'frame': images}
+            imgframe_dict = {"path": source, 'frame': images}
 
             images_update = img_process(
-                img_frame_dict, model_inference, model_data)
+                imgframe_dict, model_inference, model_data)
 
 
 def img_process(images, model_inference, model_data):
 
-    start_point = time.perf_counter()
+    # t1 = time.time()
-
     # 检测每帧图片，返回推理结果
     results = det_img(model_inference=model_inference,
                       images_frame=images['frame'],
@@ -143,10 +149,10 @@ def img_process(images, model_inference, model_data):
         
         images.update({"frame": img_save})
 
-        img_name = images_save(
+        imgname = images_save(
             images=images['frame'], save_path=model_data["save_path"])
 
-        print('sleep:', images['path'], img_name)
+        print('sleep:', images['path'], imgname)
 
         if model_data['save_path_original']:
             images_save(images=images['frame'],
@@ -156,9 +162,9 @@ def img_process(images, model_inference, model_data):
     else:
         # 没检测出来的图片是否保存
         if model_data["test_path"]:
-            img_name = images_save(
+            imgname = images_save(
                 images=images['frame'], save_path=model_data["test_path"])
-            # print('no:',images['path'],img_name)
+            # print('no:',images['path'],imgname)
 
         else:
             pass
@@ -172,11 +178,6 @@ def img_process(images, model_inference, model_data):
 
     # t2 = time.time()
 
-    end_point = time.perf_counter()
-    cost = end_point - start_point
-
-    print(f"Predicted in {cost * 1000:.2f}ms. {1.0 / cost:.2f} FPS")
-
     return images
 
 

Bank_second_part/xbank_detect_process/model_load/model_load.py

@@ -19,7 +19,7 @@ def build_option(device, backend, cache_file):
     return option
 
 
-def load_model(model_file, device, cache_file):
+def Load_model(model_file, device, cache_file):
     """
     加载模型的tensorRT引擎
     model_file：模型权重   格式：".onnx"

Bank_second_part/xbank_detect_process/utils.py

@@ -1,14 +0,0 @@
-def is_image_file(filename):
-    image_extensions = [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp"]
-    # 获取文件扩展名并转换为小写
-    file_extension = filename.split(".")[-1].lower()
-    return file_extension in image_extensions
-
-
-def is_rtsp_or_video(source):
-    if source.startswith("rtsp://"):
-        return True
-
-    video_exts = ["mp4", "avi"]
-    file_extension = source.split(".")[-1].lower()
-    return file_extension in video_exts