added trt_pose submodule

Media-4-Machine-Laboratory · Jul 21, 2021 · 9199a7b · 9199a7b
1 parent d226683
commit 9199a7b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -18,3 +18,6 @@
 	path = python/training/segmentation
 	url = https://github.com/dusty-nv/pytorch-segmentation
 	branch = master
+[submodule "plugins/pose"]
+	path = plugins/pose
+	url = https://github.com/dusty-nv/trt_pose
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -124,11 +124,11 @@ if(BUILD_EXPERIMENTAL)
 	list(APPEND inferenceIncludes ${experimentalIncludes})
 else()
 	message("-- BUILD_EXPERIMENTAL disabled")
-	file(GLOB pluginSources plugins/*.cpp plugins/*.cu)
+	file(GLOB pluginSources plugins/*.cpp plugins/*.cu plugins/pose/trt_pose/parse/*.cpp)
 endif()
 
 cuda_add_library(jetson-inference SHARED ${inferenceSources} ${pluginSources})
-
+target_include_directories(jetson-inference PRIVATE ${PROJECT_SOURCE_DIR}/plugins/pose/trt_pose/parse)
 
 # transfer all headers to the include directory
 file(MAKE_DIRECTORY ${PROJECT_INCLUDE_DIR}/jetson-inference)

diff --git a/plugins/pose b/plugins/pose