记录一次成功运行 tf2_image_retraining 。
这篇blog没有AI知识讲解,仅仅是记录运行过程。
官网地址:https://www.tensorflow.org/hub/tutorials/tf2_image_retraining
AI里面最简单的要数图像分类了,我们很少有精力去从头训练一个模型。在出色模型的基础上,会更容易有高的精确度。
1.使用VirtualBox安装Linux虚拟机,系统为:Ubuntu 18.04.4 LTS.
2.在Ubuntu安装conda.
wget https://repo.anaconda.com/archive/Anaconda3-2020.07-Linux-x86_64.sh
./Anaconda3-2020.07-Linux-x86_64.sh
然后按照提示: 一步一步操作。
为什么要安装conda呢? 简单理解就是一个方便的包管理工具(类似于pip),不过更重要的是它可以创建虚拟环境的优势。(Python2、Python3环境、TensorFlow 各个版本环境)
3. 使用conda创建虚拟环境tf_2_3_1
参考地址: https://www.jianshu.com/p/240e396e495f
(base) root@gch-pc:~# conda create -n tf_2_3_1 python=3.7.3
(base) root@gch-pc:~# source activate tf_2_3_1
4. 安装tensorflow.
pip install tensorflow==2.3.1
pip install tensorflow-hub==0.9.0
pip install pip install gast==0.3.3
pip install numpy==1.16.0
5.修改代码
由于虚拟机配置翻墙特别麻烦,直接把
https://tfhub.dev/google/imagenet/{}/feature_vector/4 对应的模型下载到本地并解压到/root/imagenet_mobilenet/
#MODULE_HANDLE ="https://tfhub.dev/google/imagenet/{}/feature_vector/4".format(handle_base)
MODULE_HANDLE = "/root/imagenet_mobilenet"# 修改为从本地加载module.
5.安装jupyter notebook并进行配置
conda install -c conda-forge jupyterlab
# 生成配置文件
#jupyter notebook --generate-config --allow-config
Writing default config to: /root/.jupyter/jupyter_notebook_config.py
# 使用ipython 命令创建访问密码
#ipython
> from notebook.auth import passwd
Out[2]: 'sha1:67c9e60bb8b6:9ffede0825894254b2e042ea597d771089e11aed'
然后修改/root/.jupyter/jupyter_notebook_config.py
c.NotebookApp.password = u'sha1:67c9e60bb8b6:9ffede0825894254b2e042ea597d771089e11aed'
c.NotebookApp.ip='*'
c.NotebookApp.open_browser = False
c.NotebookApp.port =8888 #可自行指定一个端口, 访问时使用该端口
jupyter notebook #启动
(前台启动,需要开多个窗口并且都要切换到当前虚拟环境
使用 conda activate tf_2_3_1)
通过浏览器访问http://192.168.1.204:8888
然后打开 tf2_image_retraining.ipyn文件. 点击运行
其他
1. ModuleNotFoundError 解决办法.
ModuleNotFoundError Traceback (most recent call last)
<ipython-input-3-0831fa394ed3> in <module>()
2 import os
3
----> 4 import matplotlib.pylab as plt
5 import numpy as np
6
ModuleNotFoundError: No module named 'matplotlib'
解决办法
(tf_2_3_1) root@gch-pc:~# conda install matplotlib