Чтение всех видео в папке для оптического потока

Question

Я хочу прочитать несколько видео AVI в мой код Python для оптического потока.

Это то, что я сейчас использую для одного видео.

cap = cv.VideoCapture ("029f_hap_4.8_sta_1_2.9for4sc50.avi")

Можете ли вы посоветоватьКак создать цикл для чтения всех видео, чтобы я мог запустить код для всех из них?

Спасибо!

Answer 1

Попробуйте это:

from os import listdir
from os.path import isfile
some_path = "enter your path here"

folder_contents = os.listdir(".")
for item in folder_contents:
    if os.path.isfile(item):
        do_something(item)

Answer 2

Давайте попробуем это:

import os
import cv2 as cv
dirpath = '/path/to/your/video/dir'
os.chdir(dirpath)

# assuming inside the folder, all the files are video file type
video_list = os.listdir('.')

for video in video_list:
    cap = cv.VideoCapture('./' + video)