导出onnx模型
Phil-Lin opened this issue · 5 comments
Phil-Lin commented
你好，我按照你方法修改代码并导出onnx模型（yolo7s-face.pt），然后导出模型的后处理头要比你的onnx模型要复杂很多？请问你知道这是什么问题吗？
这是你的onnx模型
这是我导出的onnx模型
we0091234 commented
你好，我按照你方法修改代码并导出onnx模型（yolo7s-face.pt），然后导出模型的后处理头要比你的onnx模型要复杂很多？请问你知道这是什么问题吗？ 这是你的onnx模型
转tensorrt能用吧，你这个
Phil-Lin commented
感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？
we0091234 commented
感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？
rv1126就用那个 ncnn里面那个export，rk的三个输出头要分开出来的，在原版的那个yolov7-face github里面有转的
we0091234 commented
感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？
Phil-Lin commented
感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？
好的，十分感谢