From 8e10c57c7e053d8789747cf1e2c5fa78f2f65cc7 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期三, 11 八月 2021 20:17:55 +0800
Subject: [PATCH] add t
---
src/detecter_tools/model.cpp | 14 ++------------
1 files changed, 2 insertions(+), 12 deletions(-)
diff --git a/src/detecter_tools/model.cpp b/src/detecter_tools/model.cpp
index f3b21a8..c548561 100644
--- a/src/detecter_tools/model.cpp
+++ b/src/detecter_tools/model.cpp
@@ -268,10 +268,10 @@
void Detecter::setOutput(int type)
{
m_OutputTensors.clear();
- printf("0-0-0-0-0-0------------------%d",type);
if(type==2)
for (int i = 0; i < 2; ++i)
{
+
TensorInfo outputTensor;
outputTensor.numClasses = CLASS_BUM;
outputTensor.blobName = "yolo_" + std::to_string(i);
@@ -323,17 +323,7 @@
{
TensorInfo outputTensor;
outputTensor.numClasses = CLASS_BUM;
- outputTensor.blobName = "yolo_" + to_string(i);
- // if (i==0)
- // {
- // outputTensor.blobName = "139_convolutional_reshape_2";
- // }else if (i==1)
- // {
- // outputTensor.blobName = "150_convolutional_reshape_2";
- // }else if (i==2)
- // {
- // outputTensor.blobName = "161_convolutional_reshape_2";
- // }
+ outputTensor.blobName = "yolo_" + std::to_string(i);
outputTensor.gridSize = (m_InputH / 32) * pow(2, 2-i);
outputTensor.grid_h = (m_InputH / 32) * pow(2, 2-i);
outputTensor.grid_w = (m_InputW / 32) * pow(2, 2-i);
--
Gitblit v1.8.0