yolov8

yolov8/001测试拼接列表.py

@@ -0,0 +1,5 @@
+
+
+l = ["a", "b", "c", "d"]
+
+print('_'.join(l))

yolov8/CMakeLists.txt

@@ -25,8 +25,8 @@ else()
   link_directories(/usr/local/cuda/lib64)
 
   # tensorrt
-  include_directories(/home/lindsay/TensorRT-8.6.1.6/include)
-  link_directories(/home/lindsay/TensorRT-8.6.1.6/lib)
+  include_directories(/home/admin-root/software/TensorRT-8.6.1.6/include)
+  link_directories(/home/admin-root/software/TensorRT-8.6.1.6/lib)
   #  include_directories(/home/lindsay/TensorRT-7.2.3.4/include)
   #  link_directories(/home/lindsay/TensorRT-7.2.3.4/lib)
 

yolov8/attendance/2025-01-08.txt

@@ -0,0 +1,12 @@
+新人打卡, 员工名：haotian ,相似度：0.99852,打卡时间2025-01-08 14:38:01.425799 
+新人打卡, 员工名：胡同同 ,相似度：0.96767,打卡时间2025-01-08 14:41:52.114084 
+新人打卡, 员工名：杨威 ,相似度：0.98566,打卡时间2025-01-08 14:57:25.434180 
+新人打卡, 员工名：张建峰 ,相似度：0.9608,打卡时间2025-01-08 14:57:42.841012 
+新人打卡, 员工名：郑俊 ,相似度：0.8994,打卡时间2025-01-08 15:17:13.444042 
+新人打卡, 员工名：马可义 ,相似度：0.90012,打卡时间2025-01-08 15:17:15.850005 
+新人打卡, 员工名：李同同 ,相似度：0.9586,打卡时间2025-01-08 15:18:17.529510 
+新人打卡, 员工名：白景辰(1) ,相似度：0.99092,打卡时间2025-01-08 15:18:34.346216 
+新人打卡, 员工名：焦军红(1) ,相似度：0.97479,打卡时间2025-01-08 15:18:39.877829 
+新人打卡, 员工名：林时波 ,相似度：0.98172,打卡时间2025-01-08 15:19:11.575465 
+新人打卡, 员工名：林凯 ,相似度：0.9983,打卡时间2025-01-08 15:27:33.302375 
+新人打卡, 员工名：于波 ,相似度：0.92919,打卡时间2025-01-08 15:27:34.106271 

yolov8/config.yaml

@@ -0,0 +1,130 @@
+engine_path: 'build/'
+
+video_config:
+
+  # 保存m3u8文件路径
+  m3u8_path: '/home/admin-root/hls_data/mid/'
+  # m3u8_path: 'mid/'
+  # 保存mp4文件路径
+  # save_path: '/home/pro/tensorrtx-master/yolov8/mp4/'
+  save_path: 'mp4/'
+
+  people_save_path: 'attendance/'
+  
+  categories : ["face", "shoe", "phone", "e-bike"]
+
+  m3u8_path_0: '/workspace/hls_data/mid/'
+
+  v0_ip: 'test243'
+  v0_channelNo: '0#'
+  v0_testclasses : [0, 1]
+  v0_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+
+  v1_ip: '192.168.21.3'
+  v1_channelNo: '1#'
+  v1_testclasses : [1]
+
+  v1_path: 'rtsp://admin:12345678a@192.168.21.3:554/Streaming/Channels/101'
+ 
+
+
+
+
+  v2_ip: '192.168.21.5'
+#  v2_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v2_channelNo: '2#'
+  v2_testclasses : [0]
+  v2_path: 'rtsp://admin:12345678a@192.168.21.5:554/Streaming/Channels/101'
+
+  v3_ip: '192.168.21.6'
+#  v3_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v3_channelNo: '3#'
+  v3_testclasses : [0]
+  v3_path: 'rtsp://admin:12345678a@192.168.21.6:554/Streaming/Channels/101'
+
+  v4_ip: '192.168.21.7'
+#  v4_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v4_channelNo: '4#'
+  v4_testclasses : [0]
+  v4_path: 'rtsp://admin:12345678a@192.168.21.7:554/Streaming/Channels/101'
+
+  v5_ip: '192.168.21.15'
+#  v5_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v5_channelNo: '5#'
+  v5_testclasses : [0]
+  v5_path: 'rtsp://admin:12345678a@192.168.21.15:554/Streaming/Channels/101'
+  
+  v6_ip: '192.168.21.30'
+#  v6_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v6_channelNo: '6#'
+  v6_testclasses : [0]
+  v6_path: 'rtsp://admin:12345678a@192.168.21.30:554/Streaming/Channels/101'
+
+  v7_ip: '192.168.21.37'
+#  v6_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v7_channelNo: '7#'
+  v7_testclasses : [0]
+  v7_path: 'rtsp://admin:12345678a@192.168.21.37:554/Streaming/Channels/101'
+
+  v8_ip: '192.168.21.50'
+  #  v6_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v8_channelNo: '8#'
+  v8_testclasses: [0]
+  v8_path: 'rtsp://admin:12345678a@192.168.21.50:554/Streaming/Channels/101'
+
+  v9_ip: '192.168.21.51'
+  #  v6_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v9_channelNo: '9#'
+  v9_testclasses: [0]
+  v9_path: 'rtsp://admin:12345678a@192.168.21.51:554/Streaming/Channels/101'
+
+  v10_ip: '192.168.21.18'
+  #  v6_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v10_channelNo: '10#'
+  v10_testclasses: [0]
+  v10_path: 'rtsp://admin:12345678a@192.168.21.18:554/Streaming/Channels/101'
+
+  v11_ip: '192.168.21.55'
+  #  v6_path: 'rtsp://10.0.0.17:8554/camera_test/2'
+  v11_channelNo: '11#'
+  v11_testclasses: [0]
+  v11_path: 'rtsp://admin:12345678a@192.168.21.55:554/Streaming/Channels/101'
+
+
+
+minioConfig:
+#  endpoint: '10.0.0.58:9000/'
+#  access_key: 'root'
+#  secret_key: '@root123456'
+#  secure: False
+#  bucket_name: 'miniotest'
+
+  #bucketName: vi
+  endpoint: '192.168.20.251:9000/'
+  access_key: 'admin'
+  secret_key: '12345678aA'
+  secure: False
+  bucket_name: 'vi-attachment'
+
+
+dataConfig:
+#  getTokenUrl: 'http://192.168.220.202/api/appsys/sso/httpheader/login/v1?username_=digital'
+  getTokenUrl: 'http://192.168.20.251/api/appsys/sso/httpheader/login/v1?username_=szls'
+#  putMessageUrl: 'http://192.168.220.200/api/edge/edgecallmanages/vi-alarm/v1'
+  putMessageUrl: 'http://192.168.20.251/api/edge/edgecallmanages/vi-alarm/v1'
+  timeInterval: 600
+#  getTokenUrl: 'http://192.168.220.202/api/appsys/sso/httpheader/login/v1/username_=digital'
+#  putMessageUrl: 'http://192.168.220.202/api/edge/edgecallmanages/vi-alarm/v1'
+
+compreface_service:
+  domain: 'http://10.0.0.202'
+  port: '8000'
+  api_key: 'ce04b456-88df-4df8-a7a2-69849111916f'
+#  api_key: 'ab77978a-cc2b-4fa0-8959-6294e856721a'
+ # api_key: '6d89a2ce-b71a-4894-96bb-03c6712e86d0'
+  # 人脸置信度，>0.9j就判断为人脸
+  det_prob_threshold: 0.99
+  # 识别图像中人脸的个数，0代表没有限制。
+  limit: 0
+
+

yolov8/include/config.h

@@ -4,7 +4,7 @@
 
 const static char* kInputTensorName = "images";
 const static char* kOutputTensorName = "output";
-const static int kNumClass = 80;
+const static int kNumClass = 4;
 const static int kBatchSize = 1;
 const static int kGpuId = 0;
 const static int kInputH = 640;

yolov8/yolov8_det_trt.py

@@ -410,7 +410,7 @@ class warmUpThread(threading.Thread):
 if __name__ == "__main__":
     # load custom plugin and engine
     PLUGIN_LIBRARY = "./build/libmyplugins.so"
-    engine_file_path = "yolov8n.engine"
+    engine_file_path = "./build/best.engine"
 
     if len(sys.argv) > 1:
         engine_file_path = sys.argv[1]
@@ -421,20 +421,7 @@ if __name__ == "__main__":
 
     # load coco labels
 
-    categories = ["person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat",
-                  "traffic light",
-                  "fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog", "horse", "sheep", "cow",
-                  "elephant", "bear", "zebra", "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase",
-                  "frisbee",
-                  "skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove", "skateboard",
-                  "surfboard",
-                  "tennis racket", "bottle", "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple",
-                  "sandwich", "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch",
-                  "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote", "keyboard",
-                  "cell phone",
-                  "microwave", "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase", "scissors",
-                  "teddy bear",
-                  "hair drier", "toothbrush"]
+    categories = ["face", "shoe", "phone", "e-bike"]
 
     if os.path.exists('output/'):
         shutil.rmtree('output/')