Skip to content

A Tensorflow Keras implementation (Graph and eager execution) of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.

Notifications You must be signed in to change notification settings

Shathe/MNasNet-Keras-Tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MNasNet

Keras (Tensorflow) Implementation of MNasNet and an example for training and evaluating it on the MNIST dataset. Check also the eager execution implementation

According to the paper: MnasNet: Platform-Aware Neural Architecture Search for Mobile

Requirement

  • Python 2.7+
  • Tensorflow-gpu 1.10

Train it

Train the MNasNet model on the MNIST dataset! just execute:

python train.py

For checking and inspecting the Mnasnet model described in the paper, execute:

python Mnasnet.py

Train it with eager execution

Train the MNasNet (eager) model on the MNIST dataset! just execute:

python train_eager.py

The eager execution implementation also outputs logs on Tensorboard. For its visualization:

tensorboard --logdir=train_log:./logs/train, test_log:./logs/test

MnasNet for... Semantic Segmentation!

In this other repository, FC-Mnasnet I added a decoder to the MnasNet architecture in order to turn it into a semantic segmentation model.

alt text

About

A Tensorflow Keras implementation (Graph and eager execution) of Mnasnet: MnasNet: Platform-Aware Neural Architecture Search for Mobile.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages