spacewalk01/tensorrt-yolov9

大佬，请问一下为什么导出的onnx模型有两个输出呢？

leayz-888 opened this issue 8 months ago · 1 comments

leayz-888 commented 8 months ago

spacewalk01 commented 8 months ago

The first one is prediction. I think the second one is loss output:
https://github.com/WongKinYiu/yolov9/blob/60a611741035d004d36fdd5c192817c8872bba1c/utils/general.py#L902