Explorar o código

test memory bug

leon hai 1 mes
pai
achega
8176b063a3
Modificáronse 3 ficheiros con 6 adicións e 6 borrados
  1. 3 3
      src/main.cpp
  2. 2 2
      src/nodes/draw/drawNode.cpp
  3. 1 1
      src/nodes/stream/streamNode.hpp

+ 3 - 3
src/main.cpp

@@ -49,12 +49,12 @@ void test_yolo()
     record_node->set_fourcc(cv::VideoWriter::fourcc('X', '2', '6', '4'));
     
     GNode::LinkNode(src_node0, infer_node);
-    GNode::LinkNode(infer_node, track_node);
-    GNode::LinkNode(track_node, draw_node);
+    // GNode::LinkNode(infer_node, track_node);
+    GNode::LinkNode(infer_node, draw_node);
     GNode::LinkNode(draw_node, record_node);
     record_node->start();
     draw_node->start();
-    track_node->start();
+    // track_node->start();
     infer_node->start();
     src_node0->start();
     getchar();

+ 2 - 2
src/nodes/draw/drawNode.cpp

@@ -87,9 +87,9 @@ void DrawNode::work()
                 std::tuple<int, int, int, int> pbox = std::make_tuple(box.left, box.top, box.right, box.bottom);
                 int x, y;
                 std::string text;
-                if (box.track_id != -1) 
+                if (box.class_id != -1) 
                 {
-                    text = str_format("%s %.2f id=%d", box.label.c_str(), box.score, box.track_id);
+                    text = str_format("%s %.2f id=%d", box.label.c_str(), box.score, box.class_id);
                 }
                 else
                 {

+ 1 - 1
src/nodes/stream/streamNode.hpp

@@ -39,7 +39,7 @@ public:
                 return;
             }
             decoder_ = FFHDDecoder::create_cuvid_decoder(
-                false, FFHDDecoder::ffmpeg2NvCodecId(demuxer_->get_video_codec()), -1, gpu_id, nullptr, nullptr, true
+                false, FFHDDecoder::ffmpeg2NvCodecId(demuxer_->get_video_codec()), 100, gpu_id, nullptr, nullptr, true
             );
             if (decoder_ == nullptr)
             {