Browse Source

update track

leon 1 tháng trước cách đây
mục cha
commit
9c27506340
3 tập tin đã thay đổi với 1 bổ sung6 xóa
  1. 0 3
      Makefile
  2. 0 1
      src/common/queue.hpp
  3. 1 2
      src/nodes/track/trackNode.cpp

+ 0 - 3
Makefile

@@ -134,9 +134,6 @@ $(objdir)/%.cu.mk : $(srcdir)/%.cu
 	@mkdir -p $(dir $@)
 	@$(nvcc) -M $< -MF $@ -MT $(@:.cu.mk=.cu.o) $(cu_compile_flags)
 
-jpg	:
-	@cd $(workdir) && rm *.jpg
-
 clean :
 	@rm -rf $(objdir) $(workdir)/$(name) $(workdir)/pro $(workdir)/*.trtmodel $(workdir)/imgs
 

+ 0 - 1
src/common/queue.hpp

@@ -64,7 +64,6 @@ public:
 
         // 检查队列是否已满,并根据策略处理
         if (queue_.size() >= max_size_) {
-            print("queue is full, size = %d\n" % queue_.size());
             switch (strategy_) {
                 case OverflowStrategy::Block:
                     // 阻塞策略:等待直到队列不再满

+ 1 - 2
src/nodes/track/trackNode.cpp

@@ -75,8 +75,7 @@ void TrackNode::work()
 
             for (const auto& track : output_stracks) {
                 const std::vector<float>& tlwh = track.tlwh;
-                int track_id = track.track_id;
-                metaData->track_boxes.emplace_back(tlwh[0], tlwh[1], tlwh[0] + tlwh[2], tlwh[1] + tlwh[3], 1.0f, track_id, track_label_);
+                metaData->track_boxes.emplace_back(tlwh[0], tlwh[1], tlwh[0] + tlwh[2], tlwh[1] + tlwh[3], track.score, track.track_id, track_label_);
             }
 
             for (auto& output_buffer : output_buffers_)