上传宏，如果用CPU会报错

6fe257ce · oscar · 3096d851 · 6fe257ce
Commit 6fe257ce authored Feb 03, 2023 by oscar
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

BaseTracker.h BaseTracker/BaseTracker.h +2 -0

No files found.
--- a/BaseTracker/BaseTracker.h
+++ b/BaseTracker/BaseTracker.h
@@ -423,7 +423,9 @@ void BaseTracker<T>::AssociateDetectionsToTrackers(const std::vector<std::vector
    }
    else
    {
+#ifdef _KF_IOU_CUDA_
        Hungarian_Cuda(iou_matrix, detections.size(), tracks.size(), association);
+#endif
    }
    /*
    std::ofstream cuda_ofs("/workspace/cuda_asso.txt", std::ios_base::out);