LTH-ObjectRecognition

The Lottery Ticket Hypothesis for Object Recognition

Sharath Girish*, Shishira R Maiya*, Kamal Gupta, Hao Chen, Larry Davis, Abhinav Shrivastava

CVPR 2021

This repository contains the code for the various experiments in our paper.

prune_voc contains the code for training directly pruned tickets and ImageNet transferred tickets on Pascal-VOC, prune_coco is for the COCO dataset and prune_imagenet contains the code for ImageNet training. Refer to the directories for installation and run instructions.

Citation

To cite this work, please use

@article{girish2020lottery,
  title={The Lottery Ticket Hypothesis for Object Recognition},
  author={Girish, Sharath and Maiya, Shishira R and Gupta, Kamal and Chen, Hao and Davis, Larry and Shrivastava, Abhinav},
  journal={arXiv preprint arXiv:2012.04643},
  year={2020}
}