1. Código fuente abierto:
- TensorFLow: https://github.com/Bisonai/mobilenetv3-tensorflow
- Documento oficial de tensorflow 1.15: https://tensorflow.google.cn/versions/r1.15/api_docs/python/tf/data/make_one_shot_iterator
2. Preparación del proyecto:
- Clon del proyecto: clon de git https://github.com/Bisonai/mobilenetv3-tensorflow
- Configuración del entorno:
- Cree un entorno virtual: conda create -n "mobilenetv3_Anne" python = 3.6
- Activación del entorno: conda activa mobilenetv3_Anne
- gpu 版 tensorflow 安装 : pip --default-timeout = 100 install tensorflow-gpu == 1.15
- scikit-image 安装 : pip --default-timeout = 100 instalar scikit-image
- instalación de tqdm: pip --default-timeout = 100 install tqdm
- tensorflow_datasets 安装 : pip install tensorflow_datasets
- instalación de ipython: conda install ipyton
- instalación de ipykernel: conda install ipykernel
- Escriba el entorno en el núcleo de jupyter: python -m ipykernel install --name moblienetv3_Anne
3. Depuración del código del proyecto
Como no conozco la versión de tensorflow del autor, la depuro directamente y modifico el informe de error directamente al uso en tensorflow 1.15.
Nota: Los documentos oficiales deben ser revisados aquí. Los documentos oficiales prevalecerán. Muchos documentos en Internet no indican qué versión se utiliza. La aplicación directa no resolverá el problema.
- Modifique build_dataset en dataset.py:
- Modifique modle.fit en train.py:
! ! ! El entrenamiento transcurrió sin problemas, pon la imagen: