使用opencv逐帧计算光流

最近写了一小段代码关于计算视频中每一帧的光流，其中遇到了一个错误，查了一些相关资料，已经解决，在此记录一下。

修改后代码

def optical_flow(file_path):
    cap = cv2.VideoCapture(file_path)
    ret,frame1 = cap.read()
    prvs = cv2.cvtColor(frame1,cv2.COLOR_BGR2GRAY)
    hsv = np.zeros_like(frame1)
    hsv[...,1] = 255
    frame_count = 1
    while (1):
        ret1,frame2 = cap.read()
        if ret1:
            next = cv2.cvtColor(frame2,cv2.COLOR_BGR2GRAY)

            #CALCULATE FLOW
            flow = cv2.calcOpticalFlowFarneback(prvs,next,None,0.5,3,15,3,5,1.2,0)
            mag,ang = cv2.cartToPolar(flow[...,0],flow[...,1])
            #x方向：flow[...,0]
            #y方向：flow[...,1]
            hsv[...,0] = ang*180/np.pi/2
            hsv[...,2] = cv2.normalize(mag,None,0,255,cv2.NORM_MINMAX)
            rgb = cv2.cvtColor(hsv,cv2.COLOR_HSV2BGR)
            cv2.imshow('frame2',rgb)
            cv2.waitKey(30)
            cv2.imwrite('D:/two_stream/flow/'+str(frame_count)+'.jpg',rgb)
            frame_count = frame_count+1
        else:
            break
    cap.release()
    cv2.destroyAllWindows()

一开始遇到的问题如下：

C:\ci\opencv_1512688052760\work\modules\imgproc\src\color.cpp:11048: error: (-215) scn == 3 || scn == 4 in function cv::cvtColor

通过查阅一些资料，发现这个报错的意思是将视频帧转化为灰度图时，视频帧本身不是彩色图像（通道数不为3或4）而指定将彩色图像转换为灰度图。一开始通过去掉循环，仅计算前两帧的光流，发现可以正常输出，因此猜测是因为计算光流时需要连续计算两帧，而最后一帧的next为空，故报错。因此加了一处修改，由于视频操作中的read（）函数返回两个值，第一个为bool值，另一个为视频帧，故通过这一参数对帧进行判断。刚接触python，深度学习没多久，有错误还望指正。