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