1、开源代码：

• TensorFLow：https://github.com/Bisonai/mobilenetv3-tensorflow
• tensorflow 1.15官方文档：https://tensorflow.google.cn/versions/r1.15/api_docs/python/tf/data/make_one_shot_iterator

2、项目准备：

• 项目克隆：git clone https://github.com/Bisonai/mobilenetv3-tensorflow
• 环境搭建：
  • 创建虚拟环境：conda create -n "mobilenetv3_Anne" python=3.6
  • 环境激活：conda activate mobilenetv3_Anne 
  • gpu版tensorflow安装：pip --default-timeout=100 install tensorflow-gpu==1.15
  • scikit-image安装：pip --default-timeout=100 install scikit-image
  • tqdm安装：pip --default-timeout=100 install tqdm
  • tensorflow_datasets安装：pip install tensorflow_datasets
  • ipython安装：conda install ipyton
  • ipykernel安装：conda install ipykernel
  • 将环境写入jupyter核中：python -m ipykernel install --name moblienetv3_Anne

3、项目代码调试

由于不清楚作者的tensorflow版本，因此直接粗暴调试，报错的地方直接修改为tensorflow 1.15里的用法即可。

注意：这里一定要查官方文档，以官方文档为准，网上很多资料并没有说明用法到底是哪个版本，直接套用解决不了问题。

• 修改dataset.py里build_dataset：
• 修改train.py里modle.fit：

！！！ 训练顺利跑起来了，放图：