spynet(三)：光流估计代码读取后缀为.flow的光流文件

tony365

已于 2022-10-19 11:03:32 修改

阅读量512

点赞数

分类专栏：图像处理算法文章标签： python 机器学习算法

于 2022-10-19 10:34:30 首次发布

本文链接：https://blog.csdn.net/tywwwww/article/details/127403108

版权

图像处理算法专栏收录该内容

93 篇文章 21 订阅

订阅专栏

9. 读取后缀为.flow的光流文件

将 .flo文件读取为 ndarray
首先读取 header，然后读取宽和高，最后读取 flow
注意fromfile的用法即可

def read_flow(filename: str) -> np.ndarray:

    f = open(filename, 'rb')

    header = f.read(4)
    if header.decode("utf-8") != 'PIEH':
        raise Exception('Flow file header does not contain PIEH')

    width = np.fromfile(f, np.int32, 1).squeeze()
    height = np.fromfile(f, np.int32, 1).squeeze()

    flow = np.fromfile(f, np.float32, width * height * 2)
    flow = flow.reshape((height, width, 2))

    return flow.astype(np.float32)