frotms / MobilenetV3-Tensorflow

the multi-GPUs implementation of mobilenet v3 in tensorflow with tf.layers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

mobilenetv3

This is a multi-GPUs Tensorflow implementation of MobileNetV3 architecture as described in the paper Searching for MobileNetV3.
paper V2 changed some layers from paper V1.
The implementation of paper V1 see branch paper_v1 in this repository for detail.

Tested on tf1.3.0, tf1.10.0, python3.5.

mobilenetv3 large

mobilenetv3 small

usage

from mobilenet_v3 import mobilenet_v3_large, mobilenetv3_small

model, end_points = mobilenet_v3_large(input, num_classes, multiplier=1.0, is_training=True, reuse=None)

model, end_points = mobilenet_v3_small(input, num_classes, multiplier=1.0, is_training=True, reuse=None)

About

the multi-GPUs implementation of mobilenet v3 in tensorflow with tf.layers

License:Apache License 2.0


Languages

Language:Python 100.0%