mobilenetv3的keras实现。
mobilenet-v3的Python项目详细描述
MobileNetv3
MobileNetV3和Lite R-ASPP语义分割的Keras实现(正在开发中)。
要求
- python 3.6
- TensorFlow GPU 1.10.0
- 路缘石2.2.4
训练模型
config/config.json
文件提供了用于培训的配置。
训练分类
数据集文件夹结构如下:
| - data/
| - train/
| - class 0/
| - image.jpg
....
| - class 1/
....
| - class n/
| - validation/
| - class 0/
| - class 1/
....
| - class n/
运行下面的命令来训练模型:
python train_cls.py
确认
感谢@fzyzcjy对本项目的帮助。
参考
@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}
}
版权所有
有关详细信息,请参见LICENSE。