19

GitHub - iamhankai/ghostnet.pytorch: [CVPR2020] Surpassing MobileNetV3: "G...

 4 years ago
source link: https://github.com/iamhankai/ghostnet.pytorch
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

readme.md

GhostNet

This repo provides Pytorch implementation of GhostNet: More Features from Cheap Operations. The TensorFlow 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 [arXiv]

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

  • Approach
ghost_module.png
  • Performance

We beat other SOTA lightweight CNNs such as MobileNetV3 and FBNet.

flops_latency.png

Citation

@article{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},
  journal={arXiv},
  year={2019}
}

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK