A Convolutional Neural Network to classify a image in 2 classes with accuracy over 86 percent over training set and 83 percent accuracy over test set.
- Keras
- numpy
- Pickle
The Convolutional Neural Network(CNN) is a feed forward network applied to image data.Here Inputs are image's 3d matrix.So,basically our objective is to find patterns and trends in similar type of images and classify them as in the same class.
This step involves moving a 3*3 martrix of feature detector across image matrix(multiply element wise) and then adding all 9 elements into one single entity of a new matrix.
We are considering a pool of 2*2 matrix along the output of previous step and either taking average of 4 entities or taking mean of 4 entites into new matrix.
we convert our matrix into a single long column vector.
Finally the column vector with important features extracted will act as a input for a fully connected layered Artificial Neural Network.