/shufflenet-v2-tensorflow

Tensorflow ShuffleNet v2 implementation

Primary LanguagePython

Shufflenet-v2-tensorflow

Tensorflow implementation of ECCV 2018 paper ShuffleNet V2. [Paper]

Support architectures

  • ShuffleNetV2 with 0.5, 1.0, 1.5, 2.0 channel multipliers

Usage

import tensorflow as tf
from net import ShuffleNetV2

if __name__ == '__main__':
    # Create input placeholder
    input = tf.placeholder(dtype=tf.float32, shape=[None, 224, 224, 3])

    # Create model
    model = ShuffleNetV2(input, 1001, model_scale=1.0, is_training=True)

    # Get model inference result
    print(model.output) # shape = [None, 1001]