search

we0091234 / yolov7-face-tensorrt

yolov7-face tensorrt
43 stars 7 forks source link

导出onnx模型 #4

Open Phil-Lin opened 1 year ago

Phil-Lin commented 1 year ago

你好，我按照你方法修改代码并导出onnx模型（yolo7s-face.pt），然后导出模型的后处理头要比你的onnx模型要复杂很多？请问你知道这是什么问题吗？ 这是你的onnx模型 image 这是我导出的onnx模型 image

we0091234 commented 1 year ago

你好，我按照你方法修改代码并导出onnx模型（yolo7s-face.pt），然后导出模型的后处理头要比你的onnx模型要复杂很多？请问你知道这是什么问题吗？ 这是你的onnx模型 image 这是我导出的onnx模型 image 转tensorrt能用吧，你这个

Phil-Lin commented 1 year ago

感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？

we0091234 commented 1 year ago

感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？

rv1126就用那个 ncnn里面那个export，rk的三个输出头要分开出来的，在原版的那个yolov7-face github里面有转的

we0091234 commented 1 year ago

感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？

image

Phil-Lin commented 1 year ago

感谢回复，我转onnx模型是准备部署在arm（瑞芯微率rv1126）上，由于后处理头操作太复杂，导致部分操作不支持，我比较困惑的是为什么我导出的onnx模型和你的差异这么多？

image

好的，十分感谢