From 3da405567797e21d0c0e90c9c4bb78bf28bb55db Mon Sep 17 00:00:00 2001 From: taojx Date: Mon, 2 Oct 2023 22:06:46 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=E8=A7=86=E9=A2=91=E5=A4=84?= =?UTF-8?q?=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../xbank_detect_process/benchmark.md | 30 +++++++++++++++++++ .../xbank_detect_process/main_process.py | 15 +++++++++- 2 files changed, 44 insertions(+), 1 deletion(-) create mode 100644 Bank_second_part/xbank_detect_process/benchmark.md diff --git a/Bank_second_part/xbank_detect_process/benchmark.md b/Bank_second_part/xbank_detect_process/benchmark.md new file mode 100644 index 0000000..d737556 --- /dev/null +++ b/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 +``` + diff --git a/Bank_second_part/xbank_detect_process/main_process.py b/Bank_second_part/xbank_detect_process/main_process.py index cf72b0b..d0aba64 100644 --- a/Bank_second_part/xbank_detect_process/main_process.py +++ b/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"]