针对分割问题,官方已经划分好了:http://cs.nyu.edu/~silberman/projects/indoor_scene_seg_sup.html
import numpy as np
import scipy.io as sio
data = sio.loadmat('splits.mat')
train_idx = data['trainNdxs']
test_idx = data['testNdxs']
train_size = len(train_idx)
test_size = len(test_idx)
partitions = [0] * (train_size + test_size)
for i in range(test_size):
idx = test_idx[i][0]
partitions[idx - 1] = 1
print partitions