python 用opencv接口把视频逐帧转化为图片

2022-09-02 21:03:19 浏览数 (1)

好了,接下来我要做一个实际的深度学习图像分割的小项目,项目内容是从一堆拍摄海面的图片中将白浪花分割出来,这个项目的分割只对白浪花感兴趣,所以最后应该是01分割,非黑即白。目前收到800G左右的数据,视频格式,每段大约50分钟。首先要做的就是从这些视频中把每一帧的图片导出来,变成图片。

这里首先有一个小插曲,我发现我的视频格式一开始虽然是mp4,但是用普通的播放器播放不了,这就很诡异。于是我下载了完美解码,果然可以播放了。不过还是很诡异,于是我用格式工厂无损的转换成mp4格式,800G的数据也跑了整整一天时间,我真佛了。

下面贴上使用python 用opencv接口把视频逐帧转化为图片的程序,当然matlab也能干这个事儿,虽然我matlab比python熟,但是以后都用python编的,所以干脆用python了。

代码语言:javascript复制
import cv2
import os
def save_img():
    video_path = 'C:/Users/86151/Desktop/test'
    videos = os.listdir(video_path)
    for video_name in videos:
        file_name = video_name.split('.')[0]
        folder_name = video_path   file_name
        os.makedirs(folder_name, exist_ok=True)
        vc = cv2.VideoCapture(video_path '/' video_name) 
        c=0
        rval=vc.isOpened()

        while rval:  
            c = c   1
            rval, frame = vc.read()
            pic_path = folder_name '/'
            if rval:
                cv2.imwrite(pic_path   str(c)   '.png', frame)
                cv2.waitKey(1)
            else:
                break
        vc.release()
        print('save_success')
        print(folder_name)
save_img()

需要特别注意:

代码语言:javascript复制
video_path = 'C:/Users/86151/Desktop/test'

0 人点赞