Skip to content
This repository was 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

4be1c5b · Aug 8, 2020

History

23 Commits
Jun 9, 2020
Dec 7, 2019
Dec 7, 2019
May 14, 2020
Aug 8, 2020

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