@@ -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();
@@ -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
@@ -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)