This repo contains source code of our work on designing efficient networks for different computer vision tasks:
- Image classification (Single/multi-label)
- Semantic segmentation
- Object detection
Table of contents
- Key highlihgts
- Supported networks
- Relevant papers
- Blogs
- Performance comparison
- Training receipe
- Citation
- License
- Acknowledgements
- Object classification on the ImageNet and MS-COCO (multi-label)
- Semantic Segmentation on the PASCAL VOC and the CityScapes
- Object Detection on the PASCAL VOC and the MS-COCO
- Supports PyTorch 1.0
- Integrated with Tensorboard for easy visualization of training logs.
- Scripts for downloading different datasets.
This repo supports following networks:
- ESPNetv2 (Classification, Segmentation, Detection)
- DiCENet (Classification, Segmentation, Detection)
- ShuffleNetv2 (Classification)
Below figure compares the performance of DiCENet with other efficient networks on the ImageNet dataset. DiCENet outperforms all existing efficient networks, including MobileNetv2 and ShuffleNetv2. More details here
Below figure compares the performance of ESPNet and ESPNetv2 on two different datasets. Note that ESPNets are one of the first efficient networks that delivers competitive performance to existing networks on the PASCAL VOC dataset, even with low resolution images say 256x256. See here for more details.
Details about training and testing are provided here.
Details about performance of different models are provided here.
Details about training and testing are provided here.
Details about performance of different models are provided here.
Details about training and testing are provided here.
If you find this repository helpful, please feel free to cite our work:
@misc{mehta2019dicenet,
Author = {Sachin Mehta and Hannaneh Hajishirzi and Mohammad Rastegari},
Title = {DiCENet: Dimension-wise Convolutions for Efficient Networks},
Year = {2019},
Eprint = {arXiv:1906.03516},
}
@inproceedings{mehta2018espnetv2,
title={ESPNetv2: A Light-weight, Power Efficient, and General Purpose Convolutional Neural Network},
author={Mehta, Sachin and Rastegari, Mohammad and Shapiro, Linda and Hajishirzi, Hannaneh},
booktitle={Proceedings of the IEEE conference on computer vision and pattern recognition},
year={2019}
}
@inproceedings{mehta2018espnet,
title={Espnet: Efficient spatial pyramid of dilated convolutions for semantic segmentation},
author={Mehta, Sachin and Rastegari, Mohammad and Caspi, Anat and Shapiro, Linda and Hajishirzi, Hannaneh},
booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
pages={552--568},
year={2018}
}
By downloading this software, you acknowledge that you agree to the terms and conditions given here.
Most of our object detection code is adapted from SSD in pytorch. We thank authors for such an amazing work.