将 .mat 文件中保存的图片转换 .png 格式的图片Python实现

单个 .mat 文件转换

import scipy.io as scio
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import os

dataFile = r'./CoNSeP/Train/Labels/train_2.mat' # 单个的mat文件
data = scio.loadmat(dataFile)
print(type(data))
# print (data['data'])
# 由于导入的mat文件是structure类型的，所以需要取出需要的数据矩阵
a=data['inst_map']
# 取出需要的数据矩阵

# 数据矩阵转图片的函数
def MatrixToImage(data):
    data = data*255
    new_im = Image.fromarray(data.astype(np.uint8))
    return new_im

new_im = MatrixToImage(a)
plt.imshow(a, cmap=plt.cm.gray, interpolation='nearest')
new_im.show()
new_im.save('train_2.png') # 保存图片

批处理

import scipy.io as scio
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import os

data_dir = './CoNSeP/Train/Labels'
path = './CoNSeP/Train/Labels/'
img_ids    =   sorted(os.listdir(data_dir))
print(len(img_ids))

for img_id in img_ids:
    dataFile =  path + img_id  # 单个的mat文件
    data = scio.loadmat(dataFile)
    print(type(data))
    # print (data['data'])
    # 由于导入的mat文件是structure类型的，所以需要取出需要的数据矩阵
    a=data['inst_map']
    # 取出需要的数据矩阵

    # 数据矩阵转图片的函数
    def MatrixToImage(data):
        data = data*255
        new_im = Image.fromarray(data.astype(np.uint8))
        return new_im

    new_im = MatrixToImage(a)
    # plt.imshow(a, cmap=plt.cm.gray, interpolation='nearest')
    # new_im.show()
    new_im.save(img_id[:-4] + '.png') # 保存图片