- 博客(4)
- 收藏
- 关注
原创 rk3588网络部署
主要流程就是先GPU训练得到待转换的.cpkt模型,再frezze得到可部署的.pb模型,以上步骤都是在自己跑模型的环境里完成的。接下来在配好环境的虚拟机里将.pb模型转换为NPU可推断的.rknn格式,最后将.rknn模型推上板子测试。
2023-08-09 16:28:50
308
1
原创 tensorflow不同模型格式的区别和含义
checkpoint、SavedModel和protobuf都是TensorFlow中常见的模型保存和加载格式,互相可以进行转换,其中.pb结尾的可能是savemodel也可能是protobuf,用来做部署的一般都是probobuf格式。
2023-08-09 15:45:39
640
1
原创 tensorflow中计算图节点和张量的命名
做模型转换的时候需要指明网络输入输出节点的名字,第一次看还挺一头雾水的,大概记录了一下tensorflow节点和张量的命名。
2023-08-09 15:05:51
194
1
原创 FINN学习笔记
finn学习前置知识ONNX创建简单的onnx模型使用onnx模型前置知识ONNXONNX(Open Neural Network Exchange)是一种针对机器学习所设计的开放式的文件格式,用于存储训练好的模型,是一个用于表示深度学习模型的标准。它使得不同的人工智能框架(如Pytorch, MXNet)可以采用相同格式存储模型数据并交互。 ONNX的规范及代码主要由微软,亚马逊 ,Facebook 和 IBM 等公司共同开发,以开放源代码的方式托管在Github上。目前官方支持加载ONNX模型并
2023-08-09 14:58:48
223
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人