search

duggalrahul / AlexNet-Experiments-Keras

Code examples for training AlexNet using Keras and Theano
MIT License
107 stars 53 forks source link

Can this project deal with datasets with over 2 classes? #5

Open zetn opened 6 years ago

zetn commented 6 years ago

Hello, I wanna ask about if this project can dealt with datasets with over 2 classes, Like cifar10?

duggalrahul commented 6 years ago

Yes. Just set the nb_classes variable to the number of classes you want. I'm not sure if AlexNet can accept inputs as small as in the CIFAR 10 dataset. You can possibly remove some pooling layers to accommodate small sized image.

Best, Rahul

G33kyKitty commented 6 years ago

Hello, why did you use nb_classes to be 2??

Gokkulnath commented 6 years ago

@G33kyKitty Because he has used Pre-trained weights for Binary Classification. (Cat/Dog)

G33kyKitty commented 6 years ago

Ok thank you @Gokkulnath ..