.flo文件读写与转换

访风景于崇阿

于 2023-07-11 13:05:40 发布

阅读量512

点赞数

分类专栏：光流估计文章标签：深度学习

本文链接：https://blog.csdn.net/starlight1231/article/details/131657851

版权

本文介绍了如何使用Python将光流图像转换为numpy数组以及将光流图像转换为可视化的图像，详细讲解了光流文件.flo与numpy及图像之间的相互转换过程。

摘要由CSDN通过智能技术生成

1. 光流图像转numpy

import numpy as np

def load_flow_to_numpy(path):
    with open(path, 'rb') as f:
        magic = np.fromfile(f, np.float32, count=1)
        assert (202021.25 == magic), 'Magic number incorrect. Invalid .flo file'
        h = np.fromfile(f, np.int32, count=1)[0]
        w = np.fromfile(f, np.int32, count=1)[0]
        data = np.fromfile(f, np.float32, count=2 * w * h)
    data2D = np.resize(data, (w, h, 2))
    return data2D


if __name__ == '__main__':
    flo = load_flow_to_numpy('frame_0001.flo')
    print(flo.shape)  # (436, 1024, 2)

2. 光流图像转图像

import numpy as np
from matplotlib.colors import hsv_to_rgb
import matplotlib.pyplot as plt

def load_flow_to_numpy(path):
    with open(path, 'rb') as f:
        magic = np.fromfile(f, np.float32, count=1)
        assert (202021.25 &#