- 安装Tensorflow，如果已安装则检查 Tensorflow版本是否大于2.3.1
文章图片
- 安装tensorflow-addons
文章图片
- 从源码安装最新的onnx(2020/12),参考git
文章图片
- 将YoloV5中的torch pt模型输出为onnx模型，参考git
文章图片
- 将onnx转化为tf的pb模型
import onnx, onnxruntime
from onnx_tf.backend import prepare
import torch
import numpy as np
import tensorflow as tfdef onnx_to_pb(onnx_path, pb_path):
onnx_model = onnx.load(onnx_path)# load onnx model
tf_rep = prepare(onnx_model)# prepare tf representation
tf_rep.export_graph(pb_path)# export the model
【机器学习实践（onnx模型转为Tensorflow2的pb模型2020）】
文章图片
测试加载模型的输出形状
文章图片
- 将保存好的tf模型做测试或预训练模型，大功告成。整个过程中注意各个包的版本符合要求。
推荐阅读
- YoloV5的原理与实现-开源TensorFlow版
- Pytorch|Pytorch optimizer.step() 和loss.backward()和scheduler.step()的关系与区别
- pandas|Pandas 速查手册
- AI|从零开始搭建自己的卷积神经网络（day.2--LeNet.5复现）
- 机器学习|翻译（AdaRNN:时间序列的自适应学习与预测：AdaRNN: Adaptive Learning and Forecasting for Time Series?）
- python|利用新闻情绪预测股市
- 【机器学习PAI实战】—— 玩转人工智能之综述
- 【提高】机器学习相关理论|【机器学习周志华】读书笔记 P2 假设空间与归纳偏好
- 机器学习|机器学习实战之朴素贝叶斯与垃圾邮件分类