python创建h5文件——图片文件夹批量制作数据集

海宝7号

于 2021-10-10 15:52:00 发布

阅读量1.1k

点赞数 1

分类专栏： python2021 文章标签： python

本文链接：https://blog.youkuaiyun.com/dongbao520/article/details/120338247

版权

本文通过python和h5py库详细介绍如何批量将图片文件夹转换为h5数据集，适合深度学习使用。文中包含完整代码示例，通过实际运行加深理解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

话不多说，代码全套完整，亲测实用NB666

只说真话，不打诳语。

import numpy as np
import matplotlib.pyplot as plt
import h5py
import os

def createData(path):
    pics = os.listdir(path)
    all_data = []
    for item in pics:
        try:
            all_data.append(plt.imread(path+item).tolist())
            # all_data.append(plt.imread(path+'/'+item).tolist())
        except Exception as pic_wrong:
            print(item+" pic wrong")
    return all_data

def createSet(hf, name, tip, data):
    hf.create_dataset(name, data=data)
    t = [[tip]*len(data)]
    hf.create_dataset(name + '_tip', data=t)

if __name__ == '__main__':
    file_train_name="D:/Pictures/zhenhua/temp001/data_train.h5"
    hf = h5py.File(file_train_name, 'w')
    all_data = createData('D:/Pictures/zhenhua/temp2/')
    createSet(hf, 'train_set_1', 1, all_dat