opencv生成flow

最新推荐文章于 2024-04-09 22:47:37 发布

bloomerOAO

最新推荐文章于 2024-04-09 22:47:37 发布

阅读量215

点赞数

文章标签： opencv 计算机视觉 python

本文链接：https://blog.csdn.net/qq_42744739/article/details/131090202

版权

def generate_flow(frame1, frame2, idx):

    frame1 = frame1.astype(np.uint8)
    frame2 = frame2.astype(np.uint8)
    prvs = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY)
    hsv = np.zeros_like(frame1)
    hsv[..., 1] = 255
    dis = cv2.optflow.createOptFlow_DeepFlow()
    next = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY)
    flow = dis.calc(prvs, next, None)
    warped = image_warp(frame2, flow)
    return warped