Upload app.py

app.py

@@ -10,6 +10,7 @@ import tempfile #创建输出临时文件夹
 import os
 from detectMotion import * #单独的运动检测
 import time
+#from matplotlib.pyplot import MultipleLocator
 
 
 #导入YoloV8模型,第一次使用会下载模型到当前文件夹当中
@@ -38,12 +39,13 @@ tracker=Sort(max_age=20,min_hits=3,iou_threshold=0.3)
 def plot_number():
     global number_of_people_in_one_frame_list, flag
     if flag == False:
-        time.sleep(5)
+        time.sleep(15)
         flag = True
-    if number_of_people_in_one_frame_list == None:
-        raise gr.Error("请先选择需要处理的视频")
     plt.close()
     fig = plt.figure()
+    plt.xlabel('Second')
+    plt.ylabel('Traffic')
+    plt.grid(linestyle='--', alpha=0.3, linewidth=2)
     plt.plot(number_of_people_in_one_frame_list)
     return fig
 
@@ -98,7 +100,8 @@ def processVideo(inputPath,codec):
     flag = False
     number_of_people_in_one_frame = 0
     number_of_people_in_one_frame_list = []
-    number_of_people=0
+    number_of_people = 0
+    sum_of_frame = 0
     cap = cv2.VideoCapture(inputPath)#从inputPath读入视频
     fps = cap.get(cv2.CAP_PROP_FPS) #获取视频的帧率
     size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)),
@@ -133,7 +136,7 @@ def processVideo(inputPath,codec):
         detections=np.empty((0, 5))#初始化运动检测
         if not(ret):#当视频全部读完，ret返回false，终止循环，视频帧读取和写入结束
             break
-        img, fig1, fig2= hisEqulColor(img)#视频增强
+        img, __, __= hisEqulColor(img)#视频增强
         #读取推理的数据
         for r in results:
             boxes=r.boxes
@@ -151,7 +154,9 @@ def processVideo(inputPath,codec):
                     #cvzone.putTextRect(img,f'{classNames[cls]}{conf}',(max(0,x1),max(30,y1)),scale=0.7,thickness=1)
                     currentArray=np.array([x1,y1,x2,y2,conf])
                     detections=np.vstack((detections,currentArray))#按行堆叠数据
-        number_of_people_in_one_frame_list.append(number_of_people_in_one_frame)
+        sum_of_frame += 1
+        if sum_of_frame % fps == 0:
+            number_of_people_in_one_frame_list.append(number_of_people_in_one_frame)
         #print(type(number_of_people_in_one_frame_list))
         number_of_people_in_one_frame = 0
         #运动检测