Merge pull request #357 from oddkiva/maint-reorganize-trt-code

MAINT: reorganize YOLO inference code.
oddkiva · Dec 11, 2023 · 7828a47 · 7828a47
2 parents a9c9dbf + 7b039ce
commit 7828a47
Show file tree

Hide file tree

Showing 48 changed files with 643 additions and 568 deletions.
diff --git a/build.py b/build.py
@@ -304,6 +304,14 @@ def build_library_docker() -> None:
         ],
         SARA_SOURCE_DIR,
     )
+    execute(
+        [
+            "docker",
+            "push",
+            f"{SARA_DOCKER_IMAGE}",
+        ],
+        SARA_SOURCE_DIR
+    )
 
 
 def build_book():

diff --git a/cpp/drafts/CMakeLists.txt b/cpp/drafts/CMakeLists.txt
@@ -13,7 +13,6 @@ add_subdirectory(ChessboardDetection)
 add_subdirectory(Calibration)
 add_subdirectory(Compute)
 add_subdirectory(MatchPropagation)
-add_subdirectory(NeuralNetworks)
 add_subdirectory(NuScenes)
 add_subdirectory(Taskflow)
 

diff --git a/cpp/drafts/NeuralNetworks/CMakeLists.txt b/cpp/drafts/NeuralNetworks/CMakeLists.txt
diff --git a/cpp/drafts/NeuralNetworks/Darknet/CMakeLists.txt b/cpp/drafts/NeuralNetworks/Darknet/CMakeLists.txt
diff --git a/cpp/drafts/NeuralNetworks/TensorRT/CMakeLists.txt b/cpp/drafts/NeuralNetworks/TensorRT/CMakeLists.txt
diff --git a/cpp/drafts/NeuralNetworks/TensorRT/InferenceExecutor.cpp b/cpp/drafts/NeuralNetworks/TensorRT/InferenceExecutor.cpp