import os
from scipy.io import loadmat
#批量
# # 原文件夹路径
# folder_path = 'E:\\X-ray\\compare\\ShanghaiTech_Crowd_Counting_Dataset\\part_A_final\\train_data\\ground_truth'
# # 遍历文件夹中的所有mat文件
# for file_name in os.listdir(folder_path):
#     if file_name.endswith('.mat'):
#         # 读取mat文件
#         mat = loadmat(os.path.join(folder_path, file_name))
#         # 获取mat文件名(不包含扩展名)
#         file_name_no_ext = os.path.splitext(file_name)[0]
#         # 新建对应的txt文件
#         txt_file = open(os.path.join(folder_path, file_name_no_ext+'.txt'), 'w')
#         gt = mat["image_info"][0, 0][0, 0][0]
#         # 将坐标写入txt文件
#         for i in range(0,len(gt)):
#             y=gt[i][1]
#             x=gt[i][0]
#             txt_file.write('{} {}\n'.format(x, y))
#         # 关闭txt文件
#         txt_file.close()

folder_path = "C:\\Users\\LENOVO\\Documents\\WeChat Files\\wxid_o3sz6ezxgqj422\\FileStorage\\File\\2023-04\\GT_IMG_50.mat"
mat = loadmat(folder_path)
txt_file = open("C:\\Users\\LENOVO\\Documents\\WeChat Files\\wxid_o3sz6ezxgqj422\\FileStorage\\File\\2023-04\\image.txt", 'w')
gt = mat["image_info"][0, 0][0, 0][0]
for i in range(0,len(gt)):
    y=gt[i][1]
    x=gt[i][0]
    txt_file.write('{} {}\n'.format(x, y))
        # 关闭txt文件
txt_file.close()

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐