Add cuda nms.

JongRok-Lee · Oct 24, 2023 · ae90c89 · ae90c89
1 parent ff22765
commit ae90c89
Show file tree

Hide file tree

Showing 7 changed files with 329 additions and 47 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -99,8 +99,7 @@ void SaveBoxPred(std::vector<pointpillar::lidar::BoundingBox> boxes, std::string
           ofs << box.h << " ";
           ofs << box.rt << " ";
           ofs << box.id << " ";
-          ofs << box.score << " ";
-          ofs << "\n";
+          ofs << box.score << "\n";
         }
     }
     else {