From 87682db7a68ae76a6f1e4443670b7ff17fe03a82 Mon Sep 17 00:00:00 2001 From: Nuzhny007 Date: Thu, 3 Oct 2024 00:11:34 +0300 Subject: [PATCH] Fix output layer index --- src/Detector/tensorrt_yolo/YoloONNXv11_instance.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/Detector/tensorrt_yolo/YoloONNXv11_instance.hpp b/src/Detector/tensorrt_yolo/YoloONNXv11_instance.hpp index ea6ea2a2..54fc6b01 100644 --- a/src/Detector/tensorrt_yolo/YoloONNXv11_instance.hpp +++ b/src/Detector/tensorrt_yolo/YoloONNXv11_instance.hpp @@ -20,8 +20,8 @@ class YOLOv11_instance_onnx : public YoloONNX const float fw = static_cast(frameSize.width) / static_cast(m_inputDims.d[3]); const float fh = static_cast(frameSize.height) / static_cast(m_inputDims.d[2]); - size_t outInd = (outputs.size() == 0) ? 0 : 1; - size_t segInd = (outputs.size() == 0) ? 1 : 0; + size_t outInd = (outputs.size() == 0) ? 1 : 0; + size_t segInd = (outputs.size() == 0) ? 0 : 1; auto output = outputs[0]; @@ -39,8 +39,8 @@ class YOLOv11_instance_onnx : public YoloONNX //std::cout << ";" << std::endl; //0: name: images, size: 1x3x640x640 - //1: name: output1, size: 1x32x160x160 - //2: name: output0, size: 1x116x8400 + //1: name: output0, size: 1x116x8400 + //2: name: output1, size: 1x32x160x160 // 25200 = 3x80x80 + 3x40x40 + 3x20x20 // 116 = x, y, w, h, 80 classes, 32 seg ancors // 80 * 8 = 640, 40 * 16 = 640, 20 * 32 = 640