/MobileNetV3

A Keras implementation of MobileNetV3.

Primary LanguagePythonMIT LicenseMIT

MobileNetV3

A Keras implementation of MobileNetV3 and Lite R-ASPP Semantic Segmentation (Under Development).

According to the paper: Searching for MobileNetV3

Requirement

  • Python 3.6
  • Tensorflow-gpu 1.10.0
  • Keras 2.2.4

Train the model

The config/config.json file provide a config for training.

Train the classification

The dataset folder structure is as follows:

| - data/
	| - train/
  		| - class 0/
			| - image.jpg
				....
		| - class 1/
		  ....
		| - class n/
	| - validation/
  		| - class 0/
		| - class 1/
		  ....
		| - class n/

Run command below to train the model:

python train_cls.py

Acknowledgement

Thank @fzyzcjy for your help in this project.

Reference

@article{MobileNetv3,  
  title={Searching for MobileNetV3},  
  author={Andrew Howard, Mark Sandler, Grace Chu, Liang-Chieh Chen, Bo Chen, Mingxing Tan, Weijun Wang, Yukun Zhu, Ruoming Pang Vijay Vasudevan, Quoc V. Le, Hartwig Adam},
  journal={arXiv preprint arXiv:1905.02244},
  year={2019}
}

Copyright

See LICENSE for details.