optflow_utils.py 文件源码

def test_flow(img1, img2): 
    gray1 = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
    gray2 = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
    flow = cv2.calcOpticalFlowFarneback(gray1, gray2, 0.5, 3, 15, 3, 5, 1.2, 0)
    cv2.imshow('flow', draw_flow(gray1, flow))
    cv2.imshow('flow HSV', draw_hsv(flow))
    cv2.imshow('warp', warp_flow(cur_glitch, flow))