I am using Digits 6.1.1 with Caffe version 0.15.14 and flavor NVIDIA. I am NOT using docker.
I am trying to use Digits to classify a healthy brain with an Alzheimer’s Disease brain but Digits is not classifying well as the validation and test each have an accuracy around 50-60%
I squashed the images which were 91x109 to 28x28 gray scale PNG files. I have 25 brains for each class. Both the validation and training images are the same size and are gray scale.
I am using the Lenet model with images that are 28x28 gray scale (which is recommended).
I have tried subtracting the mean image and no subtractions.
I am using 30 epochs with SGD and a base learning rate of 0.01 with a step down at a step size of 33% and a Gamma of 0.1