bakeup

tracking/utils/videot.py

@@ -14,38 +14,23 @@ import cv2
 # import sys
 # from scipy.spatial.distance import cdist
 
-VideoFormat = ['.mp4', '.avi']
-def video2imgs(videopath, savepath):
-    k = 0
-    have = False
-    for filename in os.listdir(videopath):
-        file, ext = os.path.splitext(filename)
-        if ext not in VideoFormat:
-            continue
-        
-        basename = os.path.basename(videopath)
-        imgbase = basename + '_' + file
-        imgdir = os.path.join(savepath, imgbase)
-        if not os.path.exists(imgdir):
-            os.mkdir(imgdir)
-
-        video = os.path.join(videopath, filename)
-        cap = cv2.VideoCapture(video)
-        i = 0
-        while True:
-            ret, frame = cap.read()
-            if not ret:
-                break
-            imgp = os.path.join(imgdir, file+f"_{i}.png")
-            i += 1
-            cv2.imwrite(imgp, frame)
-        cap.release()
-        
-        print(filename + f" haved resolved")
-        
-        k+=1
-        if k==1000:
+VideoFormat = ['.mp4', '.avi', '.ts']
+def video2imgs(videof, imgdir):
+    cap = cv2.VideoCapture(videof)
+    i = 0
+    while True:
+        ret, frame = cap.read()
+        if not ret:
             break
+        imgp = os.path.join(imgdir, f"{i}.png")
+        i += 1
+        cv2.imwrite(imgp, frame)
+        
+        if i == 400:
+            break
+    cap.release()
+    
+    print(os.path.basename(videof) + f" haved resolved")
 
 def videosave(bboxes, videopath="100_1688009697927.mp4"):
 
@@ -95,10 +80,30 @@ def videosave(bboxes, videopath="100_1688009697927.mp4"):
     cap.release()
     
 def main():
-    videopath = r'C:\Users\ym\Desktop'
-    savepath = r'C:\Users\ym\Desktop'
-    video2imgs(videopath, savepath)
-    
+    videopath = r'\\192.168.1.28\share\测试_202406\0822\A_1724314806144'
+    savepath = r'D:\badvideo'
+    # video2imgs(videopath, savepath)
+    k = 0
+    for filename in os.listdir(videopath):
+        filename = "20240822-163506_88e6409d-f19b-4e97-9f01-b3fde259cbff.ts"
+        
+        file, ext = os.path.splitext(filename)
+        if ext not in VideoFormat:
+            continue
+        
+        basename = os.path.basename(videopath)
+        imgbase = basename + '-&-' + file
+        imgdir = os.path.join(savepath, imgbase)
+        if not os.path.exists(imgdir):
+            os.mkdir(imgdir)
+
+        videof = os.path.join(videopath, filename)
+        video2imgs(videof, imgdir)
+        
+        k += 1
+        if k == 1:
+            break
+
     
     
 if __name__ == '__main__':