load_video 封装

AI算法网奇

于 2025-01-16 14:15:41 发布

阅读量270

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python基础文章标签： opencv 计算机视觉 python

本文链接：https://blog.csdn.net/jacke121/article/details/145182372

python基础专栏收录该内容

859 篇文章

订阅专栏

load_video

def load_video_frames(video_path):
    frames = []
    count = 0
    if os.path.isdir(video_path):
        
        files=glob(video_path+'/*.png')+glob(video_path+'/*.jpg')
        
        for file in files:
            img=cv2.imread(file)
            img.append(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
            frames.append(img)
            count += 1
        return frames, 25
    
    video = cv2.VideoCapture(video_path)
    fps = int(video.get(cv2.CAP_PROP_FPS))

    while True:
        ret, frame = video.read()
        if not ret:
            break
        else:
            count += 1
        frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
    return frames, fps