Skip to content
This repository has been archived by the owner on Sep 21, 2020. It is now read-only.

iamhankai/ghostnet.pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This repo only implements the demo code for GhostNet. Please move to the official repo for more details.

GhostNet

This repo provides demo Pytorch implementation of CVPR 2020 paper GhostNet: More Features from Cheap Operations. The TensorFlow/PyTorch implementation with pretrained model is available at here.

Requirements

The code was verified on Python3, PyTorch 1.0+.

Usage

Usage example:

import torch
from ghost_net import ghost_net

model = ghost_net(width_mult=1.0)
input = torch.randn(32,3,224,224)
y = model(input)
print(y)

Introduction of GhostNet

GhostNet: More Features from Cheap Operations. CVPR 2020. [arXiv]

By Kai Han, Yunhe Wang, Qi Tian, Jianyuan Guo, Chunjing Xu, Chang Xu.

  • Approach
  • Performance

GhostNet beats other SOTA lightweight CNNs such as MobileNetV3 and FBNet.

Citation

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