- 博客(2)
- 收藏
- 关注
RSS订阅原创 验证pytorch转onnx模型预测精度是否正确
在训练完pytorch模型后，往往根据实际项目的需要将torch模型转为onnx模型，而转换完后可能会存在一个问题，即自己转换的onnx模型是否转换正确，模型精度是否与torch模型一致。本文以retinaface训练的torch模型为例，默认torch模型已经转换为onnx模型，验证该转换onnx模型是否与torch模型预测精度一致。具体代码示例如下：import onnxruntime as ortimport cv2import numpy as npimport torchdef i
2020-07-29 19:54:02
4894
4
原创 pytorch之Retinaface测试图片报错小结
pytorch训练的Retinaface模型预测，将图片等比例缩放至网络输入大小，进行图片测试时，有时会出现如下错误：通过排查发现是因为网络输出预测的大小与PriorBox计算输出的大小不一致导致，而造成该问题则是图片等比例缩放的大小不是cfg['steps’]的整数倍导致的,故需对resize完的图片再进行填充处理，使之能够整除cfg['steps’]。具体解决如下：# pad input image to avoid unmatched shape problemimg, pad_par
2020-07-29 19:46:16
469
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人