Home

Awesome

HAF in PyTorch

Official Code Release for Learning Hierarchy Aware Features for Reducing Mistake Severity <br/> Ashima Garg, Depanshu Sani, Saket Anand. <br />

European Conference on Computer Vision (ECCV 2022)

Citations

If you find this paper useful, please cite our paper:

@inproceedings{garg2022learning,
  title={Learning Hierarchy Aware Features for Reducing Mistake Severity},
  author={Garg, Ashima and Sani, Depanshu and Anand, Saket},
  booktitle={European Conference on Computer Vision},
  pages={252--267},
  year={2022},
  organization={Springer}
}

Proposed Approach

<div align="center"> <img src="imgs/teaser_hist.png"/> </div>

Installation

Clone this repository

$ git clone https://github.com/07Agarg/HAF.git
$ cd HAF

Dataset Preparation

Refer to Repository: Making Better Mistakes

Hierarchies

Refer to Repository: Making Better Mistakes

Using the Code for Training

The experiments in the paper are contained in the folder experiments/ dataset wise. For CIFAR-100

bash experiments/train/cifar-100/cross-entropy.sh

For iNaturalist-19

bash experiments/train/inat/cross-entropy.sh

For tiered-imagenet

bash experiments/train/tieredimagenet/cross-entropy.sh

For testing

Refer to the code repository: CRM-making-better-mistakes

Link To Trained Models

Download HAF final trained models from the link and use the above repository for testing.

Acknowledgements

This codebase is borrowed from making-better-mistakes

Contact

If you have any suggestion or question, you can leave a message here or contact us directly at ashimag@iiitd.ac.in