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.