Tencent / tencent-ml-images

Largest multi-label image database; ResNet-101 model; 80.73% top-1 acc on ImageNet
Other
3.06k stars 515 forks source link

a question about the num_class #58

Open genghuan2005 opened 5 years ago

genghuan2005 commented 5 years ago

Hi, thank you for sharing, I am interested to know that why the class_num is set to be 1000 in the flag.py. Is the given model in checkpoints just suitable for predicting the top 1000 classes of the dictionary?