CapsNet image classification

Requirements:

1. Dataset with train, valid, test folders and use folder to specify the target class.
2. Python venv with Pytorch Cuda.

Split into 2 sections (jupyter files):

1. Train part including fine tuning using optuna.
2. Evaluation on test dataset including explicable AI part.