import os
import scipy.io as sio
# 定义包含MAT文件的文件夹路径
folder_path = '你要处理的mat文件所在的文件夹路径'
# 获取文件夹中的所有MAT文件
mat_files = [f for f in os.listdir(folder_path) if f.endswith('.mat')]
# 创建一个空标签字典,用于存储每个MAT文件的标签
labels_dict = {}
# 循环处理每个MAT文件
for mat_file in mat_files:
# 构建MAT文件的完整路径
mat_file_path = os.path.join(folder_path, mat_file)
# 读取MAT文件
mat_data = sio.loadmat(mat_file_path)
# 获取MAT文件中的数据
data = mat_data['data'] # 替换为实际的数据变量名
# 通过输入来手动添加标签
label = input(f"为MAT文件 '{mat_file}' 添加标签 (1 或 0): ")
# 将标签存储在标签字典中,以文件名作为键
labels_dict[mat_file] = int(label)
# 选择要保存标签文件的文件夹路径
output_folder = '你要保存标签文件的目标文件夹路径'
output_path = os.path.join(output_folder, 'labels.txt')
# 打印每个MAT文件的标签
for mat_file, label in labels_dict.items():
print(f"MAT文件: {mat_file}, 标签: {label}")
# 保存标签到指定文件夹中的文件
with open(output_path, 'w') as file:
for mat_file, label in labels_dict.items():
file.write(f"{mat_file}: {label}\n")
print(f"标签已保存到 '{output_path}' 文件中.")
二分类问题手动给mat数据打标签
最新推荐文章于 2024-07-12 11:32:03 发布