agatan/tf-mobilenetv2

MobileNetV2 written in tensorflow, training with eager mode and estimator API

MobileNetV2 in TensorFlow

This is a TensorFlow implementation of MobileNetV2: Inverted Residuals and Linear Bottlenecks.

Training

main.py is an example training script using eager execution. main_estimator.py is using estimator API.