leon hai 1 mes
pai
achega
96485ad5c2
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)
                 {