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}
}