Skip to content

WillCheung2016/MobileNet_V2_Keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 

Repository files navigation

MobileNet V2 using Keras

Author of the script: Shengdong Zhang

Email: [email protected]

This is the beta implementation of Mobilenet V2(https://128.84.21.199/pdf/1801.04381.pdf) using Keras.

Because there are still some contradiction in the model description part in the paper, this script is implemented based on the best understanding of the script author. Updates will be made as soon as it is ready or the paper is updated.

MobileNet v2 is the next version of MobileNet v1 with big improvement. Instead of directly using depthwise convolution + 1x1 convolution structure, it implements inverted residual block structure by first expanding input data into a larger dimension and then applying 3x3 depthwise convolution plus 1x1 convolution bottlenet structure to decrease dimension. Based on the experiments in the paper and my personal experience, this structure does help gradients pass through the deep network, which leverage the gradient vanishing problem.

Currently modification needed if you want to use the script for images with small sizes like CIFAR10 or MNIST. ImageNet pretrained weights will be released as soon as it is available.

Reference

Releases

No releases published

Packages

No packages published

Languages