isi-vista / unified-io-inference

Apache License 2.0
0 stars 0 forks source link

Score Imagenet predictions #19

Closed yash-reddy closed 1 year ago

yash-reddy commented 1 year ago

Closes #16

Add the script to score classification predictions made on the Imagenet dataset. Scoring requires two additional files - LOC_synset_mapping.txt and *LOC__solution.csv**, both of which can be found in /nas/gaia02/data/paper2023/imagenet

Predictions can be scored using the following command - python score/imagenet.py --predictions PREDICTIONS_FILE --gold-labels PATH_TO_SOLUTION --synsets PATH_TO_SYNSETS