leon 1 month ago
parent
commit
96485ad5c2
1 changed files with 1 additions and 1 deletions
  1. 1 1
      src/nodes/infer/inferNode.cpp

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

@@ -33,7 +33,7 @@ void InferNode::work()
             }
             auto det_result = model_->forward(tensor::cvimg(image));
 
-            if (std::holds_alternative<std::vector<data::BoxArray>>(det_result)) {
+            if (std::holds_alternative<data::BoxArray>(det_result)) {
                 auto result = std::get<data::BoxArray>(det_result);
                 for (auto& r : result)
                 {