Skip to content

ShuffleNet model

Jump to bottom
Luigi edited this page Oct 1, 2018 · 2 revisions

X. Zhang, X. Zhou, M. Lin, and J. Sun, “Shufflenet: An extremely efficient convolutional neural network for mobile devices,” in Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2018.

@InProceedings{Zhang_2018_CVPR,
author = {Zhang, Xiangyu and Zhou, Xinyu and Lin, Mengxiao and Sun, Jian},
title = {ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices},
booktitle = {Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2018},
organization={IEEE}
}

Source: ShuffleNet repo

  • shufflenet(num_classes=1000, pretrained='imagenet')

Back to GitHub Page

Clone this wiki locally