add log for -DTRT_10

csrc/detect/end2end/CMakeLists.txt

@@ -32,6 +32,7 @@ print_var(TensorRT_LIBRARIES)
 print_var(TensorRT_INCLUDE_DIRS)
 print_var(TensorRT_LIB_DIR)
 if (TensorRT_VERSION_MAJOR GREATER_EQUAL 10)
+    message(STATUS "Build with -DTRT_10")
     add_definitions(-DTRT_10)
 endif ()
 

csrc/detect/end2end/include/common.hpp

@@ -7,8 +7,6 @@
 #include "NvInfer.h"
 #include "filesystem.hpp"
 #include "opencv2/opencv.hpp"
-#include <sys/stat.h>
-#include <unistd.h>
 
 #define CHECK(call)                                                                                                    \
     do {                                                                                                               \

csrc/detect/end2end/include/yolov8.hpp

@@ -5,7 +5,7 @@
 #define DETECT_END2END_YOLOV8_HPP
 #include "NvInferPlugin.h"
 #include "common.hpp"
-#include "fstream"
+#include <fstream>
 using namespace det;
 
 class YOLOv8 {

csrc/detect/end2end/main.cpp

@@ -1,9 +1,9 @@
 //
 // Created by ubuntu on 1/20/23.
 //
-#include "chrono"
 #include "opencv2/opencv.hpp"
 #include "yolov8.hpp"
+#include <chrono>
 
 namespace fs = ghc::filesystem;