yolov5-face 转化ONNX时如何得到与推理相同的输出格式？

转化ONNX时如何得到与推理相同的输出格式？

Open suncheng-s opened this issue 2 years ago • 10 comments

转化ONNX时如何得到与推理相同的输出格式？目前转化为ONNX时，输出为三个特征图的结果。怎么修改能得到和推理时一样的输出结果呢？如下图所示。

我修改了model.model[-1].export = False，输出格式相同，但是提示错误：

请问该如何解决？

Nov 01 '21 02:11 suncheng-s

你的右图是yolov5的onnx,如果你想得到类似的结构，修改yolo.py

Nov 01 '21 08:11 derronqi

@suncheng-s 已支持，请参考https://github.com/deepcam-cn/yolov5-face/pull/108#issue-1083085258

Dec 17 '21 10:12 bobo0810

@suncheng-s 已支持，请参考#108（评论）

可以提供正确转onnx 和推理onnx的代码吗感谢期待您的回复

Apr 15 '22 03:04 sssssshf

@suncheng-s 已支持，请参考#108（评论）

可以提供正确转onnx 和推理onnx的代码吗感谢期待您的回复

已经合并到主分支了，直接用就可以。

Apr 15 '22 05:04 bobo0810

您的意思是说yolo5face 的代码库已经更新过？下载最新的git项目需要重新训练然后在运行转onnx就没问题了吗

Apr 15 '22 05:04 sssssshf

您的意思是说yolo5face 的代码库已经更新过？下载最新的git项目需要重新训练然后在运行转onnx就没问题了吗

目前权重可以直接转为onnx 后处理只需要NMS等了。

Apr 15 '22 05:04 bobo0810

您的英文 olo5face 的代码库更新过吗？

当前权重可以转为onnx后处理只需要NMS等直接了。

我现在转onnx的过程会报错，代码已经按照上述的两个地方作了修改： model.model[-1].export = False # set Detect() layer export=True model.model[-1].export_cat = True 报错如下： aten::view(Tensor(a) self, int[] size) -> (Tensor(a)): Expected a value of type 'Tensor' for argument 'self' but instead found type 'None'. :

Apr 15 '22 05:04 sssssshf

yolov5-face yolov5-face copied to clipboard

转化ONNX时如何得到与推理相同的输出格式？

yolov5-face
yolov5-face copied to clipboard