'''
将CSV文件格式转换为.h5文件。类别SB,SS,SD,SN,
'''
import numpy as np
import h5py
import json
import os
def read_file():
# 从第二行开始读取数据
# 每一行的第一列为序号,最后一列为类别
# 每个样本有690个数据,即光谱有690个波段
# 生成 hdpy5 文件格式
data = []
classes = []
file_list = []
num_data = []
with open("salmon.csv", 'r') as f:
next(f)
# 通过循环将文件数据转换成列表
for line in f.readlines():
line = line.strip().split(',')
# 将列表中字符串格式转为float类型
new_data = list(map(eval, line[1:-1]))
# file_list 为数据列表
# classes 为数据对应标签
file_list.append(new_data)
if line[-1]=='"SD"':
classes.append(0)
if line[-1]=='"SN"':
classes.append(1)
if line[-1]=='"SS"&#
将csv格式的数据制作h5py格式数据
最新推荐文章于 2022-12-12 23:59:48 发布