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.
原创
2022-05-22 11:57:31 ·
459 阅读 ·
0 评论