Build for Linux with CMake

.gitignore

@@ -2,4 +2,6 @@
 x64/
 *.user
 *.pt
-*.mp4
+*.mp4
+build/
+CMakeLists.txt~

CMakeLists.txt

@@ -0,0 +1,16 @@
+cmake_minimum_required(VERSION 3.18.4)
+
+project(siam-trackers LANGUAGES CXX)
+set(CMAKE_CXX_STANDARD 20)
+
+find_package(OpenCV REQUIRED)
+find_package(Torch REQUIRED)
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
+
+add_definitions(-DOPENCV)
+
+add_executable(${PROJECT_NAME} main.cpp)
+
+target_sources(${PROJECT_NAME} PRIVATE Tracker.cpp TrackerSiamMask.cpp TrackerSiamRPNPP.cpp)
+
+target_link_libraries(${PROJECT_NAME} PRIVATE ${OpenCV_LIBS} ${TORCH_LIBRARIES})

main.cpp

@@ -3,13 +3,18 @@
 #include <opencv2/opencv.hpp>
 #include "TrackerSiamRPNPP.h"
 #include "TrackerSiamMask.h"
+
+#ifdef _WIN32
 #include "windows.h"
+#endif
 
 const cv::Scalar COLOR_BLUE(255, 0, 0);
 
 int main(int argc, char **argv) {
+    #ifdef _WIN32
     _putenv_s("OMP_NUM_THREADS", "8");
     LoadLibraryA("torch_cuda.dll");
+    #endif
 
     // TODO: Should we `eval` these modules?
     // SiamMask