pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

[FEATURE] Validation labels txt or csv file

Open IgorKasianenko opened this issue 3 years ago • 1 comments

I have trained model using timm distributed_train.sh, but I have 0.1 validation accuracy on imagenet. Images in validation folder are all placed in root directory going like ILSVRC2012_val_00000001.JPEG ... ILSVRC2012_val_00050000.JPEG.

Describe the solution you'd like I would like to have parameter in validate.py file similar to '--real-labels' that will take csv or txt file with structure like

ImageId,Class
ILSVRC2012_val_00000001,n01751748

IgorKasianenko avatar Jan 11 '22 16:01 IgorKasianenko

run https://github.com/PatrickHua/EasyImageNet/blob/main/valprep.sh in the validation folder of your imagenet copy (the one featuring 50k files) and everything will run smoothly from this point

aliutkus avatar Feb 15 '22 13:02 aliutkus