Ubuntu训练tflite模型

1. 安装Anaconada

 **1.1   查看电脑内核(终端输入)**
uname -m
#返回x86_64
#比如我的电脑是x86_64的,就在清华源镜像网站下载所需要的文件
#镜像网址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive

在这里插入图片描述
不建议下载最新版(最新版可能会有bug)。
################################################################
下载后进入文件夹,解压文件
使用bash命令编译安装,例如:

bash Anaconda3-2021.11-Linux-x86_64.sh 
#一直按yes就可以了

#安装完成测试conda指令
conda
#如果没有报错就安装好了,如果有问题,应该是路径没有输出,

1.2 提示 “conda:未找到命令” 的解决办法

#终端输入
sudo gedit ~/.bashrc
#在文本末尾添加路径
export PATH="/home/用户名/anaconda3/bin:$PATH"	#用户名改为自己的用户名
source ~/.bashrc 								#立即更新环境变量

1.3 更换conda下载源为清华源

conda config --add channels 'https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/'   
conda config --add channels 'https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/'  
conda config --set show_channel_urls yes

1.4 Anaconda安装完毕
1.5 conda常用指令

conda update conda #检查更新当前conda
conda env list#查看虚拟环境
conda create -name your_env_name python=X.X			#创建虚拟环境,your_env_name 是你要创建的名字
#我需要创建的是tensorflow2的简称tf2,python3.7即
conda create -name tf2 python=3.7
#选择yes就行
#环境创建完毕,然后激活
conda activate tf2

conda remove -name your_env_name(虚拟环境名称) --all  #删除虚拟环境中全部的
conda remove --name your_env_name  package_name  	#删除虚拟环境中全部的某个包

2.安装tensorflow及opencv

2.1在创建的tf2环境中使用

pip3 install tensorflow==2.4.1			#安装tf2.4.1
pip3 install --upgrade pip				#升级pip
pip3 install opencv-python==4.1.0.25 	#安装opencv

2.2测试是否安装完成

python

#下面的语句一句话一句话复制
import tensorflow as tf
import cv2 as cv

print(tf.__version__)		#查看tensorflow版本
print(tf.__path__)			#查看tensorflow路径
print(cv.__version__)		#查看opencv版本
print(cv.__path__)			#查看opencv路径

3.安装TensorFlow Object Detection API

3.1下载模型源码
在主文件夹创建tensorflow文件夹,并将下载的文件名修改为models,存放到tensorflow文件夹中。

3.2安装protoc

3.2.1下载protoc源码

3.2.2解压

sudo cp bin/protoc /usr/bin/protoc

cd tensorflow/models/research
protoc object_detection/protos/*.proto --python_out=.#将proto文件编译为py文件

运行完成后,可以检查object_detection/protos/文件夹,如果每个proto文件都成了对应的以py为后缀的python源码,就说明编译成功了。
如图:
在这里插入图片描述
3.3 添加路径,并安装
进入models/research/object_detection/packages路径下,将tf2中的setup.py复制到models/research/下

cd models/research/slim
python setup.py build
python setup.py install
sudo gedit ~/.bashrc
#在文本末尾添加
export PYTHONPATH=$PYTHONPATH:/home/用户名/tensorflow/models/research:/home/用户名/tensorflow/models/research/slim
cd ../
python setup.py build
python setup.py install
pip install .#下载所需的所有包

3.4 测试是否安装完成

python object_detection/builders/model_builder_test.py

说一点,我安装的是tensorlow—cpu,所以无需cuda等

4.TensorFlow2 CPU训练及导出pb并转tflite

5.树莓派4B 安装Ubuntu 18.04 部署深度学习框架

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值