这里是接着上一篇博文写的。
1.dataSet, 建立一个用于存储和格式化读取训练数据的类。将数据集分开为train, test.建立标签。
因为借鉴的博客是theno做的backend。
#重新格式化和标准化
# 本案例是基于thano的,如果基于tensorflow的backend需要进行修改
X_train = X_train.reshape(X_train.shape[0], 1, self.img_size, self.img_size)/255.0
X_test = X_test.reshape(X_test.shape[0], 1, self.img_s