News
2020/11/10 The code of TinyNet (NeurIPS 2020) has been release at MindSpore Model Zoo.
2020/10/31 GhostNet+TinyNet achieves better performance. See details in our NeurIPS 2020 paper: arXiv.
2020/09/24 We release GhostNet models for more vision tasks on MindSpore Hub and MindSpore Model Zoo.
2020/06/10 GhostNet is included in PyTorch Hub.
This repo provides pretrained models and inference code for TensorFlow and PyTorch:
- Tensorflow: ./ghostnet_tensorflow with pretrained model.
- PyTorch: ./ghostnet_pytorch with pretrained model.
- We also opensource code on MindSpore Hub and MindSpore Model Zoo.
For training, please refer to tinynet or timm.
This repo provides pretrained models and inference code for PyTorch:
- PyTorch: ./tinynet_pytorch with pretrained model.
- We also opensource training code on MindSpore Model Zoo.
GhostNet: More Features from Cheap Operations. CVPR 2020. [Paper] [arXiv]
By Kai Han, Yunhe Wang, Qi Tian, Jianyuan Guo, Chunjing Xu, Chang Xu.
GhostNet beats other SOTA lightweight CNNs such as MobileNetV3 and FBNet.
@inproceedings{ghostnet,
title={GhostNet: More Features from Cheap Operations},
author={Han, Kai and Wang, Yunhe and Tian, Qi and Guo, Jianyuan and Xu, Chunjing and Xu, Chang},
booktitle={CVPR},
year={2020}
}
@article{tinynet,
title={Model Rubik’s Cube: Twisting Resolution, Depth and Width for TinyNets},
author={Han, Kai and Wang, Yunhe and Zhang, Qiulin and Zhang, Wei and Xu, Chunjing and Zhang, Tong},
journal={Advances in Neural Information Processing Systems},
volume={33},
year={2020}
}
This repo provides the TensorFlow/PyTorch code of GhostNet. Other versions and applications can be found in the following:
- timm: code with pretrained model
- Darknet: cfg file, and description
- Gluon/Keras/Chainer: code
- Paddle: code
- Bolt inference framework: benckmark
- Human pose estimation: code
- YOLO with GhostNet backbone: code
- Face recognition: cavaface, FaceX-Zoo