python 读取 mat 格式，corel 5k数据集

2018-01-10

在corel 5k数据集中，图像标签存储形式为mat格式，可是电脑上没有装matlab，所以利用python进行读取，再转成自己需要的格式。
以corel5k_test_annot.mat为例

import scipy.io as sio
import numpy as np

# matlab文件名
matfn = u'D:\data\Corel5K-master\真正的Corel5k数据集(带标注,训练与测试集)\corel5k_test_annot.mat'

# load数据
data = sio.loadmat(matfn)

print(len(data))  # print结果为 4，具体如下data格式所述
data = data['annot2']
print(len(data)) # print结果为 499，即测试用例共499个
print(len(data[0]))  # print结果为 260，即标签共260个

其中，data里的格式为：

{
  '__header__': b'MATLAB 5.0 MAT-file, Platform: PCWIN, Created on: Fri Oct 25 22:30:59 2013',
  '__version__': '1.0',
  'annot2': array([[0, 1, 1, ..., 0, 0, 0],
       [0, 0, 1, ..., 0, 0, 0],
       [0, 0, 0, ..., 0, 0, 0],
       ...,
       [0, 0, 1, ..., 0, 0, 0],
       [0, 0, 1, ..., 0, 0, 0],
       [0, 0, 0, ..., 0, 0, 0]], dtype=uint16),
  '__globals__': []
}