Update app.py

app.py

@@ -185,14 +185,14 @@ class VideoTransformer(VideoTransformerBase):
     def __init__(self):
         self.snapshots = []  # 存储捕获的快照
         self.last_capture_time = time.time()  # 上次捕获时间
-        self.capture_interval = 0.5  # 每0.5秒捕获一张快照
+        self.capture_interval = 1  # 每0.5秒捕获一张快照
 
     def transform(self, frame):
         """从摄像头流捕获单帧图像，并转换为 PIL Image。"""
         img = frame.to_ndarray(format="bgr24")
         current_time = time.time()
-        # 每隔 capture_interval 秒捕获一张快照，直到捕获20张
-        if current_time - self.last_capture_time >= self.capture_interval and len(self.snapshots) < 20:
+        # 每隔 capture_interval 秒捕获一张快照，直到捕获5张
+        if current_time - self.last_capture_time >= self.capture_interval and len(self.snapshots) < 5:
             img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
             self.snapshots.append(Image.fromarray(img_rgb))
             self.last_capture_time = current_time
@@ -312,7 +312,7 @@ def photo_detection_page():
 
 def real_time_detection_page():
     st.title("实时视频检测")
-    st.write("程序在一分钟内捕获20张快照进行图片分类，首先判定是否吸烟。若检测到吸烟的快照超过2次，则展示年龄与性别分类结果。")
+    st.write("程序在一分钟内捕获5张快照进行图片分类，首先判定是否吸烟。若检测到吸烟的快照超过2次，则展示年龄与性别分类结果。")
 
     # 创建用于显示进度文字和进度条的占位容器
     capture_text_placeholder = st.empty()
@@ -328,7 +328,7 @@ def real_time_detection_page():
     image_placeholder = st.empty()
     audio_placeholder = st.empty()
 
-    capture_target = 20
+    capture_target = 5
 
     if ctx.video_transformer is not None:
         classification_result_placeholder = st.empty()
@@ -402,7 +402,7 @@ def real_time_detection_page():
                     classification_text_placeholder.text("分类进度: 分类完成！")
                     classification_progress.progress(100)
 
-                time.sleep(1)
+                time.sleep(5)
                 classification_progress_placeholder.empty()
                 classification_text_placeholder.empty()
                 capture_text_placeholder.empty()