The code about ImageNet dataset?

jiawei-ren / BalancedMetaSoftmax-Classification

[NeurIPS 2020] Balanced Meta-Softmax for Long-Tailed Visual Recognition

https://github.com/jiawei-ren/BalancedMetaSoftmax

Other

135 stars 26 forks source link

The code about ImageNet dataset? #4

Closed blue-blue272 closed 3 years ago

blue-blue272 commented 3 years ago

Can you upload the code about the training of ImageNet?

Thanks!

jiawei-ren commented 3 years ago

The configuration files for ImageNet-LT dataset have been provided in config/ImageNet-LT.

We have not tried our algorithm on the original ImageNet dataset, since ImageNet-LT has been a more popular benchmark in existing long-tailed classification works. You may change the option dataset: ImageNet_LT to dataset: ImageNet in the configuration file to use the original ImageNet data list. Do let us know if you have any result on the original ImageNet dataset!

jiawei-ren commented 3 years ago

I will close the issue for now. Feel free to reopen it if you have any further questions.