onnx
文章平均质量分 71
nodototao
这个作者很懒,什么都没留下…
展开
-
.h5转onnx,获取onnx中间层特征值
综上所述,整体代码和命令的意义是:首先,使用Keras加载一个预训练的深度学习模型,查看其结构摘要,并将其保存为TensorFlow SavedModel格式。因为近些年工作中基本都是使用pytorch框架,tensorflow框架还是最初的时候使用过一段时间,现在一个项目中给到的训练后的模型是基于tensorflow keras保存的.h5文件。这里,模型被保存为TensorFlow的SavedModel格式,这是一种更为通用和可移植的格式,特别适用于在不同平台或工具之间共享模型。原创 2024-06-15 10:28:06 · 604 阅读 · 1 评论 -
retinaface中pytorch转onnx遇到的问题
说明:基于retinaFace源码,pytorch转onnx遇到的问题。**问题一:opset版本**解决方法 :添加opset_version=11参数torch_out = torch.onnx.export(net, inputs, output_onnx, export_params=True, verbose=True, input_names=input_names, output_names=output_names, opset_version=11)问题二:opencv版原创 2021-12-16 17:13:03 · 2214 阅读 · 0 评论