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

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

posted @   昊月光华  阅读(4)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· 单线程的Redis速度为什么快?
点击右上角即可分享
微信分享提示