jingweimo / food-image-classification-

ten-class food images and classification based on cnn in python
9 stars 6 forks source link

Regarding the Network of the Model #2

Open vikash512 opened 7 years ago

vikash512 commented 7 years ago

Hi, your code is perfect and thank you for the developing such an Code ,it's very useful and also getting right prediction. Now I am trying to do some small changes in the code as I have 200 classes of food dataset prepared by me and total images are approximately 130000 images of train and validation.I need to classify the multi label classification with confidence score or probability .Is i need to change the network sizes in code,can you please suggest how can perform multi label classification?I have performed with 50 classes of images with 50epochs with your code but i got the accuracy 21%. So i am assuming it s not giving the correct predictions with probability levels.How can i resolve it?