search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.21k stars 1.27k forks source link

obj_list (string array containing class labels) contains empty strings #349

Open pocaha opened 4 years ago

pocaha commented 4 years ago

Hey zylo117,

first and foremost, thank you for the great work. My question: the obj_list array in the coco.yml file (in projects folder) contains all the 80 classes of the coco dataset2017. However, there are also 11 empty strings. Don't they affect the model performance?

pocaha commented 4 years ago

@mnslarcher ?

zylo117 commented 4 years ago

there are 90 classes in the pretrained weights.

pocaha commented 4 years ago

Coco2017 has 80 classes, ImageNet has way more than 90 classes, With which datasets were the weights pretrained? Also the number of the classes from the pretraining dataset don't have to match with the actual training set as else we would not use ImageNet pretrained weights for EfficientNet. Im training from scratch. By the way which pretraining do you mean? EfficientNet pretraining or EfficientDet pretraining?