How can I use my own dataset and fit it to your code

[HirparaAmit]

Dear Agron

I am following ur work which is in extra_capsnets.ipynb. You use MNIST data set for your work. My question is how I can use my own dataset and fit it to your coding.

I have an issue with following line: X_batch, y_batch = mnist.train.next_batch(batch_size)

Please help me regarding this.

Thanks and Regards Amit