每秒提取文件夹中所有视频的帧

Question

我试图每秒提取一个视频帧，同时在一个文件夹中有多个视频。我让它适用于像这样的 1 个视频，但我认为我搞乱了所有视频的循环。以下是 1 个有效视频的代码。

import cv2
pathOut = r"C:/Users/Me/Out/"
vidcap = cv2.VideoCapture(r'C:\Me\Desktop\test.mp4');
count = 0
success = True
while success:
    success,image = vidcap.read()
    print('read a new frame:',success)
    if count%30 == 0 :
         cv2.imwrite(pathOut + 'frame%d.jpg'%count,image)
    count+=1

对于所有视频的循环，我都是这样制作的。

import os
import cv2
pathOut = r"C:/Users/Me/Out/"
count = 0
success = True
counter = 1
listing = os.listdir(r'C:/Users/Me/videos/train')
for vid in listing:
    vid = r"C:/Users/Me/videos/train/"+vid
    cap = cv2.VideoCapture(vid)
    count = 0
    counter += 1
    while success:
        success,image = cap.read()
        print('read a new frame:',success)
        if count%30 == 0 :
             cv2.imwrite(pathOut + 'frame%d.jpg'%count,image)
        count+=1

我的视频循环似乎不起作用，因为它只拍摄一个视频。然后它状态为 false，可能是因为没有剩余帧，但我不知道如何将其推进到下一个视频。我想我需要做一些小的调整，有人知道具体是什么吗？

Answer 1

这似乎是一个小逻辑错误：一旦第一个视频将 success 设置为 false，它将永远不会再次设置为 true，并且所有视频的所有后续 while 循环都将被跳过。尝试将您的程序更改为：

import os
import cv2
pathOut = r"C:/Users/Me/Out/"
count = 0
counter = 1
listing = os.listdir(r'C:/Users/Me/videos/train')
for vid in listing:
    vid = r"C:/Users/Me/videos/train/"+vid
    cap = cv2.VideoCapture(vid)
    count = 0
    counter += 1
    success = True
    while success:
        success,image = cap.read()
        print('read a new frame:',success)
        if count%30 == 0 :
             cv2.imwrite(pathOut + 'frame%d.jpg'%count,image)
        count+=1