hazerd论文中mat格式深度图转提取并转npy文件

处理与转换MAT深度数据到.npy格式

最新推荐文章于 2024-12-19 17:57:53 发布

原创最新推荐文章于 2024-12-19 17:57:53 发布 · 454 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#计算机视觉 #人工智能

图像处理同时被 2 个专栏收录

27 篇文章

订阅专栏

图像去雾加雾

14 篇文章

订阅专栏

该代码遍历指定目录下的所有MAT文件，读取其中的imDepth字段，然后将这些深度数据保存为.npy文件。文件路径信息存储在result_depth列表中，并且每个文件的名称加上.npy后缀。

import numpy
import scipy.io as scio
import os
dataFile = r'C:\Users\Administrator\Desktop\test\hazerd\data\depth'

result_depth = []
for root, _, fnames in sorted(os.walk(dataFile)):
    for fname in fnames:
        path = os.path.join(root, fname)
        result_depth.append(path)

for i in result_depth:
    data = scio.loadmat(i)
    f,n = os.path.split(i)
    name = os.path.join(f,n+'.npy')
    depth = data['imDepth']
    numpy.save(name,depth)