测试视频处理

Bank_second_part/xbank_detect_process/benchmark.md

@@ -0,0 +1,30 @@
+# 样本
+
+- 30秒玩手机视频
+
+- /home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi
+
+- 20fps
+
+# Before 89746c13ca
+
+```
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:37.985733 563
+Predicted in 11.54ms. 86.66 FPS
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:38.003012 564
+Predicted in 11.39ms. 87.82 FPS
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:38.020148 565
+Predicted in 11.40ms. 87.71 FPS
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:38.037283 566
+Predicted in 11.46ms. 87.29 FPS
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:38.054427 567
+Predicted in 11.40ms. 87.71 FPS
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:38.071553 568
+Predicted in 11.78ms. 84.92 FPS
+/home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi 2023-10-02 21:47:38.089046 569
+
+...
+
+总共处理时间: 10.877327028021682s
+```
+

Bank_second_part/xbank_detect_process/main_process.py

@@ -31,6 +31,7 @@ def data_load(args):
                                  cache_file=model_data["model_cache"])
 
     if rtsp_or_video_source:
+        start_point = time.perf_counter()
 
         cap = cv2.VideoCapture(source)
 
@@ -40,7 +41,14 @@ def data_load(args):
                 ret, frame = cap.read()
 
                 if not ret:
-                    continue  # 如果未成功读取到视频帧，则继续读取下一帧
+                    
+                    # 流媒体不稳定,
+                    # 如果未成功读取到视频帧，则继续读取下一帧
+                    if source.startswith('rtsp'):
+                        continue  
+                    else:
+                        break
+
                 print(source,datetime.today(), i)
                 # if source == 'rtsp://admin:@192.168.10.18':
 
@@ -65,6 +73,11 @@ def data_load(args):
 
         cap.release()
 
+        end_point = time.perf_counter()
+        print(f"视频处理时间: {(end_point - start_point)}s")
+
+        exit(0)
+
     if dir_source:
 
         img_ext = [".jpg", ".JPG", ".bmp"]