leon 1 月之前
父节点
当前提交
edbdf83abe
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/nodes/infer/inferNode.cpp

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

@@ -47,7 +47,7 @@ void InferNode::work()
                 printf("model is nullptr\n");
                 printf("model is nullptr\n");
                 continue;
                 continue;
             }
             }
-            auto det_result = model_->forward(tensor::cvimg(image));
+            auto det_result = model_->forward(tensor::cvimg(image), image.cols, image.rows, 0.0f, 0.0f);
 
 
             if (std::holds_alternative<data::BoxArray>(det_result)) 
             if (std::holds_alternative<data::BoxArray>(det_result)) 
             {
             {