sandesh612 / vegetable-classification-using-cnn Goto Github PK
View Code? Open in Web Editor NEWCreated cnn model using rescaling layer , 3 convolution layer with relu activation each of them having max pooling layer. Used adam optimizer and sparse categorical crossentropy for loss function. Did data augmentation as the giving data was less for training and testing the model.