Add files

.gitignore

@@ -0,0 +1,70 @@
+*.o
+*.a
+*.dSYM
+*.csv
+*.out
+*.png
+*.so
+*.exe
+*.dll
+*.lib
+*.dylib
+*.pyc
+*~
+mnist/
+data/
+caffe/
+grasp/
+images/
+opencv/
+convnet/
+decaf/
+submission/
+cfg/
+temp/
+build/darknet/*
+build_*/
+ninja/
+ninja.zip
+vcpkg_installed/
+!build/darknet/YoloWrapper.cs
+.fuse*
+*.weights
+*.db
+*.engine
+build/*
+build/*.cmake
+build/*.ninja
+build/*.txt
+build/*.json
+build/CMakeFiles/
+build/detect_cuda_compute_capabilities.cu
+build/.ninja_deps
+build/.ninja_log
+build/Makefile
+*/vcpkg-manifest-install.log
+build.log
+__pycache__/
+
+# OS Generated #
+.DS_Store*
+ehthumbs.db
+Icon?
+Thumbs.db
+*.swp
+
+# IDE generated #
+.vs/
+.vscode/
+
+# Managed by CMake
+src/version.h
+
+# Build artifacts
+lib/
+share/
+include/darknet/
+uselib
+uselib_track
+darknet
+vcpkg/

CMakeLists.txt

@@ -0,0 +1,29 @@
+cmake_minimum_required(VERSION 3.1) 
+project(lightNet-TRT VERSION 1.0)            
+enable_language(CUDA)
+
+find_package(OpenMP REQUIRED)
+set(CMAKE_CXX_COMPILIER "/usr/bin/g++")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17 -O3 -Wno-write-strings ${OpenMP_CXX_FLAGS} -fopenmp")
+#set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++17 -O0 -g -Wno-write-strings")
+#set(CMAKE_CXX_FLAGS_DEBUG "-O0 -ggdb -g")
+set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,-rpath -Wl,$ORIGIN")
+
+
+
+include_directories(/usr/local/cuda/include)
+find_package(CUDA REQUIRED)
+
+find_package(OpenCV REQUIRED)
+
+file(GLOB_RECURSE sources modules/*.hpp modules/*.cpp modules/*.h modules/*.cu extra/*.h module/preprocess.cu)
+add_library(detector SHARED ${sources})
+target_include_directories(detector PRIVATE extra/ modules/ ${OpenCV_INCLUDE_DIRS} ${CUDA_TOOLKIT_ROOT_DIR}/include)
+target_link_libraries(detector nvinfer nvinfer_plugin nvparsers gflags "stdc++fs")
+
+
+add_executable(lightNet-TRT samples/sample_detector.cpp ${sources})
+
+
+target_include_directories(lightNet-TRT PRIVATE modules/ extra/)
+target_link_libraries(lightNet-TRT  ${OpenCV_LIBS} nvinfer nvinfer_plugin nvparsers gflags boost_system boost_filesystem "stdc++fs" "stdc++")

configs/calibration_images.txt

@@ -0,0 +1,100 @@
+/home/umedan/dataset/BDD100K/images/100k/train/a1cae59e-38fccb63.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1b95f02-9b7efc64.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a172319c-3c355f01.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a190b8b1-d2ab4fb0.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a16471ef-ade2f942.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1f0c1ea-7448b7e3.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1b084bb-51329423.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1abe365-c3c638d8.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a13ce748-dfe70d69.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1efc30f-22cf14b5.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1acfdb8-6a938192.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1a4cb3f-1d6bd2cc.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a14b280f-756a617e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a118fb8c-efaea076.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a188e945-124b917e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1bee171-9421bb0a.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1c807b7-1bb9f9ac.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1853fee-5aeb7eac.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1c4c708-6a84d574.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1b895b8-ba49d6ec.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a18c14fc-3dd8ed7e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a14f8bc3-0f647fea.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1cf97bc-1d6fd9f2.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a17759bf-9bfe862c.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a12d8ff4-c8c0b1fc.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a110e9ed-9e1e288a.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a17b173a-f3e0c23c.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1a148cd-3650fac9.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1140112-bc993046.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1fac5cb-4798356c.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a148b47e-9efca019.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1149db1-8a15ba7d.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1b6bc44-3ecbbbe6.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1eef013-c19c159d.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1429aa6-d76a5785.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a14b36f2-27e4006e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a16daafe-01fc534b.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a110e9ed-fe91fa73.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a17241f2-83fdd18a.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a12c1d69-f526cfce.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1c46767-7a89c619.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1f981a6-a89bbd22.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1ab3be8-e97bda9e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1d21246-bc6b4e33.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1a42dcd-daa8d404.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1b7d976-771facab.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a19b9103-6959a7c9.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1eafb96-ce45e383.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1d82b67-9b2e7926.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1186e90-f66c3825.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1d82b67-85d9ac13.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a13ca7b2-109284e2.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1250ed4-bac9d101.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a16b31f6-7f7ea97e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a166fa25-e4f7d66e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1f981a6-56298e6e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1acfdb8-391d8281.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1c525ce-0e965af8.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1a4d977-6285bc92.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a12b0fdc-7ad0c862.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a128c310-6d0a8695.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1cf97bc-502a2997.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1054b37-8303b687.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a19ba1f1-ebfffdb6.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1274137-58f53abb.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1088d0a-2b4275d0.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1831070-c69e2219.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a13eb5c9-6b3d6acd.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a12f1d14-1c6a524a.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a110f1cd-30064b30.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a114f8b0-8e49ddcb.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a175fd53-018d8c7e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1b7d976-827815ed.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1597863-fe66afc5.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a13a816f-18abb933.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1a148cd-6037c795.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a120d1d0-9b6cdc96.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a10dd550-b8e057ea.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a17759bf-e1123836.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1429aa6-f1ab409d.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a179876b-4c48b230.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a16b31f6-3913a385.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a13a4782-64c5a43a.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a110e9ed-8182e2f6.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a14089eb-02be8f6a.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1110075-4cd0ab61.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1c82385-8d7d4e08.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a10cf4cc-52ee3a7f.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1224f71-e7fd1aad.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a165ad95-78c1dc92.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a10131c2-bf94f303.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1575ae0-c1415768.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1140112-51665c1e.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1bee171-89109f99.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a15e2423-c25e9d50.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1bb1731-9e0a8e80.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a110e9ed-897d7a30.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1555d5a-91feac58.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a13e7efd-bb2d6090.jpg
+/home/umedan/dataset/BDD100K/images/100k/train/a1acd9c2-20af91bd.jpg