20200622:mobilenetv3-tensorflow实战(一):项目准备和代码调试

    技术2022-07-11  88

    1、开源代码:

    TensorFLow:https://github.com/Bisonai/mobilenetv3-tensorflowtensorflow 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.15scikit-image安装:pip --default-timeout=100 install scikit-imagetqdm安装:pip --default-timeout=100 install tqdmtensorflow_datasets安装:pip install tensorflow_datasetsipython安装:conda install ipytonipykernel安装:conda install ipykernel将环境写入jupyter核中:python -m ipykernel install --name moblienetv3_Anne

    3、项目代码调试

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

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

    修改dataset.py里build_dataset:修改train.py里modle.fit:

    !!! 训练顺利跑起来了,放图:

    Processed: 0.010, SQL: 9