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

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