Python基于OpenCV实现视频数据切割为图像数据

忘是亡心i 2022-09-11 08:24 104阅读 0赞

在图像处理领域里面OpenCV是一款很强大的开源软件了，最近正好有一个需求就是需要将给定的视频切分为单帧的图像数据，之后对图像数据进行处理，这里我选择的是OpenCV来完成这项工作，代码实现如下所示：

def video2Frames(vedio_path='a.mp4',picDir='images/',delta=10,h=416,w=416):
        '''
        视频数据切割为图像数据
        '''
        if not os.path.exists(picDir):
            os.makedirs(picDir)
        videoCapture = cv2.VideoCapture()
        videoCapture.open(vedio_path)
        fps=videoCapture.get(cv2.CAP_PROP_FPS)  #帧率
        frames=videoCapture.get(cv2.CAP_PROP_FRAME_COUNT)  #总帧数
        if delta:
            print('fps: ', fps, 'frames: ', frames, 'delta: ', delta)
            for i in range(int(frames)):
                ret,frame=videoCapture.read()
                #图像缩放处理
                try:
                    frame=cv2.resize(frame,(h,w),interpolation=cv2.INTER_NEAREST)
                    if i%delta==0:
                        cv2.imwrite(picDir+str(i)+'.png',frame)
                except Exception as e:
                    print('Exception: ',e)
        else:
            print("fps=", int(fps), "frames=", int(frames))
            for i in range(int(frames)):
                ret,frame=videoCapture.read()
                #图像缩放处理
                try:
                    frame=cv2.resize(frame,(h,w),interpolation=cv2.INTER_NEAREST)
                    cv2.imwrite(picDir+str(i)+'.png',frame)
                except Exception as e:
                    print('Exception: ',e)

整体处理还是比较清晰的，这里我就不再多做解释了，相应的注释已经放到代码对应的部分里面去了。