Can I chnage the number of output classes?

[GooDrYu]

Can I chnage the number of output classes? e.g. I Want to reduce the default 150 ouput classes to 10

[haiderasad]

i also want to know this kindly respond

[Raffael1998]

Hi, I am very interested too. I want to pick ~20 classes from the 150. (i want to focus on urban/ road related classes) One idea would be to take the output probability vector of each class and just take the max probability of the 20 chosen classes. But this solution is not the best, a better way would be to retrain the model and limit to those 20 classes. I would also like to limit this training to the urban related training pictures. Any idea on how to achieve that ? Thanks