keras data generator 调试

在写了代码：https://blog.youkuaiyun.com/u013249853/article/details/90408122

之后，我们希望能够看到，每次到底生成了什么

因为如果你有数据扩增的代码，你就会想知道我生成的对不对，能不能调试

，那么简单直观的调试代码如下

[a,b] = training_generator.__getitem__(1)
validation_generator = DataGenerator(partition['validation'], **params)
img = a[0]
seg = a[1]
plt.imshow(img[0,:,:,:])
plt.savefig('img.png')
plt.imshow(seg[0,:,:])
plt.savefig('seg.png')
b1 = argmax(b,axis=3)
plt.imshow(b1[0,:,:])
plt.savefig('lab.png')

然后你需要到你的代码存储目录下，查看生成的这三张图片。如果你有多张图片的话