add LISRD

CMakeLists.txt

@@ -10,9 +10,17 @@ set(onnxruntime_INCLUDE_DIRS
   ${onnxruntime_INSTALL_PREFIX}/include/onnxruntime
   ${onnxruntime_INSTALL_PREFIX}/include/onnxruntime/core/session
 )
-
 find_library(onnxruntime_LIBS NAMES onnxruntime PATHS /usr/local/lib)
 
+option(ENABLE_TORCH "ENABLE LibTorch Inference" ON)
+if(ENABLE_TORCH)
+  add_definitions(-DENABLE_TORCH)
+  set(Torch_DIR "/path/to/libtorch")
+  find_package(Torch REQUIED)
+else()
+  message(STATUS "DISABLE LibTorch Inference")
+endif()
+
 find_package(CUDA QUIET)
 
 if(CUDA_FOUND AND USE_GPU)

examples/CMakeLists.txt

@@ -195,3 +195,22 @@ target_include_directories(loftr
   PUBLIC
     ${OpenCV_INCLUDE_DIRS}
 )
+
+# ---------------------------------------------------------
+
+add_executable(lisrd
+  ${CMAKE_CURRENT_LIST_DIR}/LISRD.cpp
+  ${CMAKE_CURRENT_LIST_DIR}/SuperPoint.cpp
+  ${CMAKE_CURRENT_LIST_DIR}/LISRDApp.cpp
+)
+
+target_link_libraries(lisrd
+  PUBLIC
+    ${PROJECT_NAME}
+    ${OpenCV_LIBS}
+)
+
+target_include_directories(lisrd
+  PUBLIC
+    ${OpenCV_INCLUDE_DIRS}
+)

examples/LISRD.cpp

@@ -0,0 +1,25 @@
+/**
+ * @file    SuperPoint.cpp
+ *
+ * @author  btran
+ *
+ */
+
+#include "Lisrd.hpp"
+#include "Utility.hpp"
+
+namespace Ort
+{
+void Lisrd::preprocess(float* dst, const unsigned char* src, const int64_t targetImgWidth,
+                            const int64_t targetImgHeight, const int numChannels) const
+{
+    for (int i = 0; i < targetImgHeight; ++i) {
+        for (int j = 0; j < targetImgWidth; ++j) {
+            for (int c = 0; c < numChannels; ++c) {
+                dst[c * targetImgHeight * targetImgWidth + i * targetImgWidth + j] =
+                    (src[i * targetImgWidth * numChannels + j * numChannels + c]);
+            }
+        }
+    }
+}
+}

examples/LISRD.hpp

@@ -0,0 +1,32 @@
+/**
+ * @file    SuperPoint.hpp
+ *
+ * @author  btran
+ *
+ */
+
+#pragma once
+
+#include <opencv2/opencv.hpp>
+#include <ort_utility/ort_utility.hpp>
+#include <vector>
+
+namespace Ort
+{
+class Lisrd : public OrtSessionHandler
+{
+public:
+  static constexpr int64_t IMG_H = 480;
+  static constexpr int64_t IMG_W = 640;
+  static constexpr int64_t IMG_CHANNEL = 3;
+
+using OrtSessionHandler::OrtSessionHandler;
+
+void preprocess(float* dst,                     //
+                const unsigned char* src,       //
+                const int64_t targetImgWidth,   //
+                const int64_t targetImgHeight,  //
+                const int numChannels) const;
+
+};
+}  // namespace Ort