From 58a60434b3e026a2db3e280f1b9214697bc2735a Mon Sep 17 00:00:00 2001
From: triplemu <gpu@163.com>
Date: Tue, 4 Jun 2024 13:34:09 +0800
Subject: [PATCH] add log for -DTRT_10

---
 csrc/detect/end2end/CMakeLists.txt     | 1 +
 csrc/detect/end2end/include/common.hpp | 2 --
 csrc/detect/end2end/include/yolov8.hpp | 2 +-
 csrc/detect/end2end/main.cpp           | 2 +-
 4 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/csrc/detect/end2end/CMakeLists.txt b/csrc/detect/end2end/CMakeLists.txt
index 2cdd232..53a7463 100644
--- a/csrc/detect/end2end/CMakeLists.txt
+++ b/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 ()
 
diff --git a/csrc/detect/end2end/include/common.hpp b/csrc/detect/end2end/include/common.hpp
index 0edfb28..dedf40f 100644
--- a/csrc/detect/end2end/include/common.hpp
+++ b/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 {                                                                                                               \
diff --git a/csrc/detect/end2end/include/yolov8.hpp b/csrc/detect/end2end/include/yolov8.hpp
index 049a087..795eb0e 100644
--- a/csrc/detect/end2end/include/yolov8.hpp
+++ b/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 {
diff --git a/csrc/detect/end2end/main.cpp b/csrc/detect/end2end/main.cpp
index 83d86cf..f714335 100644
--- a/csrc/detect/end2end/main.cpp
+++ b/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;