修改bug

infer.py

@@ -7,6 +7,7 @@
 '''
 import cv2
 import torch
+import time
 import numpy as np
 from ultralytics import YOLO
 from shapely.geometry.polygon import Polygon
@@ -243,6 +244,8 @@ if __name__ == "__main__":
         result = instance(image)
         if len(result)>0 and time.time()-posttime>30:
             videoTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
+            filename  = videoTime + ".jpg"
+            filenameori = videoTime + "det.jpg"
             logger.info(videoTime)
             logger.info(result)
             for res in result:
@@ -255,7 +258,7 @@ if __name__ == "__main__":
                         'classIndex': '8',
                         'ip': '172.19.152.231',
                         'videoTime': videoTime,
-                        'videoUrl': video}
+                        'videoUrl': stream.stream_url}
             files = [
                             ('file', (filename, content, 'image/jpeg')),
                             ('oldFile', (filenameori, contentori, 'image/jpeg')),

stream.py

@@ -11,6 +11,7 @@ class StreamCapture:
         self.ip                 = ip
         self.reconnect_interval = reconnect_interval
         self.cap                = None
+        self.stream_url         = None
 
     def get_stream_url(self):
         data = {
@@ -18,6 +19,7 @@ class StreamCapture:
             "ip":self.ip
         }
         url = requests.post(url='http://172.19.152.231/open/api/operate/previewURLs',data=data).json()['msg']
+        self.stream_url = url
         return url
     
     def get_frame(self):