UserWarning: You are trying to export the model with onnx:Upsample for ONNX opset version 9

最新推荐文章于 2024-11-26 10:08:23 发布

原创最新推荐文章于 2024-11-26 10:08:23 发布 · 594 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

试错专栏收录该内容

1 篇文章

订阅专栏

这段代码展示了如何将预训练的RetinaFace模型（基于Mobilenet0.25配置）加载到PyTorch中，并使用torch.onnx.export将其导出为ONNX格式。在导出过程中，通过设置opset_version=11来解决可能的警告。

部署运行你感兴趣的模型镜像

retinanet = RetinaFace(cfg=cfg_mnet, phase='eval', pre_train=False).eval()
state_dict = torch.load('model_data/Retinaface_mobilenet0.25.pth')
retinanet.load_state_dict(state_dict)

retinanet.eval();


dummy_input=torch.randn(1,3,640,640)
torch.onnx.export(retinanet, dummy_input,'retinanet.onnx', export_params=True,verbose=False,
                  output_names=['output0'], input_names=['input0'],opset_version=11)

解决警告：加opset_version=11

您可能感兴趣的与本文相关的镜像