leon 2 месяцев назад
Родитель
Сommit
156bf39509
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      src/nodes/infer/inferNode.cpp

+ 2 - 1
src/nodes/infer/inferNode.cpp

@@ -32,7 +32,8 @@ void InferNode::work()
             // int h = rand() % (height - y);
             // metaData->boxes.push_back(data::Box(x, y, x + w, y + h, 0.9, 0));
 
-            auto res = model_->forward(tensor::cvimg(image), image.cols, image.rows, 0.0f, 0.0f);;
+            // auto res = model_->forward(tensor::cvimg(image), image.cols, image.rows, 0.0f, 0.0f);
+            auto res = model_->forward(tensor::cvimg(image));
             for (auto& r : res)
             {
                 metaData->boxes.push_back(r);