MobileNetV3

by xiaochus

xiaochus /MobileNetV3

A Keras implementation of MobileNetV3.

155 Stars 57 Forks Last release: Not found MIT License 19 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

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.

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.