Java Deep Learning Cookbook

There's more...

Our CNN has an accuracy of around 50%. We trained our neural network using 396 images across 4 categories. For an i7 processor with 8 GB of RAM, it will take 15-30 minutes to complete the training. This can vary depending on the applications that are running parallel to the training instance. Training time can also change depending on the quality of the hardware. You will observe better evaluation metrics if you train with more images. More data will contribute toward better predictions. And, of course, it demands extended training time.

Another important aspect is to experiment with the number of hidden layers and subsampling/convolution layers to give you the optimal results. Too many layers could result in overfitting, ...