Hello World AI In TensorFlow2.0

记录一次成功运行 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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值