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 @ 2022-05-22 11:57  昊月光华  阅读(10)  评论(0)    收藏  举报  来源