Skip to content

MobileNet Implementation of Torch Implemenetation of "MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications"

License

Notifications You must be signed in to change notification settings

dimkastan/torch-mobilenet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

torch-mobilenet

MobileNet implementation for torch based on Pytorch Equivalent: https://github.com/marvis/pytorch-mobilenet

Description

This is a Torch implementation of Mobilenet architecture.

TODO

a. Add bias once the nn library solves the issue with Bias on DepthWiseSeparable convolution.
b. Add training on imagenet and food101.
c. Add pretrained model and demo

Important Note

Training is very slow. In particular, it takes about 300msec to process a batch of 10 images (30msec/image) on a NVIDIA GTX-1070.

Please feel free to contact me for any comments or suggestions.

About

MobileNet Implementation of Torch Implemenetation of "MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications"

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages