CrackForest数据集| .mat转为.png

2023-12-26 16:50:15

数据集下载

CrackForest数据集
https://github.com/cuilimeng/CrackForest-dataset

转化代码

input_folder = 'xxxxxxxxx'

# 输出文件夹路径
output_folder = 'xxxxxxxxx'

# 遍历文件夹中的.mat文件
for filename in os.listdir(input_folder):
    if filename.endswith('.mat'):
        print(filename)
        # 构建完整的文件路径
        mat_filepath = os.path.join(input_folder, filename)

        # 从.mat文件中加载数据
        mat = scipy.io.loadmat(mat_filepath)

        np_seg = mat['groundTruth'][0][0][0]
        (y, x) = np.where(np_seg == 2)
        np_seg[y, x] = 255
        (y, x) = np.where(np_seg == 1)
        np_seg[y, x] = 0

        cv2.imwrite(output_folder+filename+'.png', np_seg)

文章来源:https://blog.csdn.net/holly_Z_P_F/article/details/135224219
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。