leon преди 4 седмици
родител
ревизия
3f4f8b95bf
променени са 2 файла, в които са добавени 1 реда и са изтрити 2 реда
  1. 1 1
      src/nodes/base/base.hpp
  2. 0 1
      src/nodes/infer/inferNode.cpp

+ 1 - 1
src/nodes/base/base.hpp

@@ -116,7 +116,7 @@ static inline void LinkNode(const std::shared_ptr<BaseNode> &front,
     const std::shared_ptr<BaseNode> &back, int queue_size = 100, OverflowStrategy strategy = OverflowStrategy::Block) 
 {
     PLOGI.printf("Link Node %s --> %s", front->get_name().c_str(), back->get_name().c_str());
-    auto queue = std::make_shared<SharedQueue<std::shared_ptr<meta::MetaData>>>();
+    auto queue = std::make_shared<SharedQueue<std::shared_ptr<meta::MetaData>>>(queue_size, strategy);
     back->add_input_buffer(front->get_name(), queue);
     front->add_output_buffer(back->get_name(), queue);
 }

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

@@ -37,7 +37,6 @@ void InferNode::handle_data(std::shared_ptr<meta::MetaData>& meta_data)
         auto result = std::get<data::BoxArray>(det_result);
         for (auto& r : result)
         {
-            std::cout << r.seg_mask.size() << std::endl;
             meta_data->boxes.push_back(r);
         }
     }