datang complite

.gitignore

-/build
+build
 /test_video
 /.vscode
 *.mp4
 *.md~
 /yolov3/configs/models
-/data
\ No newline at end of file
+/data

CMakeLists.txt

@@ -13,8 +13,8 @@ set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,-rpath -Wl,$ORIGIN")
 # tensorrt
 #include_directories(/usr/include/aarch64-linux-gnu/)
 #link_directories(/usr/lib/aarch64-linux-gnu/)
-include_directories(/home/zszt/whl/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/include)
-link_directories(/home/zszt/whl/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/lib)
+include_directories(/usr/local/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/include)
+link_directories(/usr/local/whl/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/lib)
 
 
 add_subdirectory(yolov3)

yolov3/CMakeLists.txt

@@ -20,8 +20,8 @@ find_package(CUDA REQUIRED)
 # tensorrt
 #include_directories(/usr/include/aarch64-linux-gnu/)
 #link_directories(/usr/lib/aarch64-linux-gnu/)
-include_directories(/home/zszt/whl/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/include)
-link_directories(/home/zszt/whl/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/lib)
+include_directories(/usr/local/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/include)
+link_directories(/usr/local/TensorRT-7.0.0.11/targets/x86_64-linux-gnu/lib)
 
 #opencv
 #set(OpenCV_DIR /opt/opencv-4.1.2/share/OpenCV)
@@ -40,9 +40,9 @@ target_include_directories(gpu PRIVATE modules/ ${OpenCV_INCLUDE_DIRS} ${CUDA_TO
 target_link_libraries(gpu "stdc++fs" ${OpenCV_LIBS} nvinfer nvinfer_plugin nvcaffe_parser )
 
 #sample
-#add_executable(yolo-trt samples/sample_detector.cpp)				  
-#target_include_directories(yolo-trt PRIVATE modules/ extra/ ${OpenCV_INCLUDE_DIRS})
-#target_link_libraries(yolo-trt ${OpenCV_LIBS} gpu opencv_dnn)
+add_executable(yolo-trt samples/sample_detector.cpp)				  
+target_include_directories(yolo-trt PRIVATE modules/ extra/ ${OpenCV_INCLUDE_DIRS})
+target_link_libraries(yolo-trt ${OpenCV_LIBS} gpu opencv_dnn)
 
 
 

yolov3/modules/ajson.hpp

-#pragma once
+#ifndef _JUEFX_AJSON_HPP_
+#define _JUEFX_AJSON_HPP_
 
 /*
 __ _____ _____ _____
@@ -2247,3 +2248,4 @@ namespace ajson\
 };\
 }
 
+#endif
\ No newline at end of file

yolov3/samples/sample_detector.cpp

@@ -66,7 +66,7 @@ using namespace std;
 
 int main(int argc, char** argv)
 {
-    InitCnrt(false);
+    //InitCnrt(false);
     string file = "/media/6542bc52-3d7a-4bea-b9b6-f9343ca8560f/yolov3/configs/calibration_images.txt";
     ifstream infile; 
     infile.open(file.data());   //将文件流对象与文件连接起来 
@@ -123,4 +123,4 @@ int main(int argc, char** argv)
  
     // delete input_mats;
     // Destory();
-}
\ No newline at end of file
+}