diff --git a/pkg/onnx/onnx.cpp b/pkg/onnx/onnx.cpp
index fe9620f..692782e 100644
--- a/pkg/onnx/onnx.cpp
+++ b/pkg/onnx/onnx.cpp
@@ -1,3 +1,4 @@
+#include <fstream>
 #include <onnxruntime_c_api.h>
 #include <vector>
 #include <string>
diff --git a/pkg/onnx/onnx.go b/pkg/onnx/onnx.go
index e2c3bfd..3173dfe 100644
--- a/pkg/onnx/onnx.go
+++ b/pkg/onnx/onnx.go
@@ -1,8 +1,8 @@
 package onnx
 
 /*
-#cgo CXXFLAGS: -std=c++17 -I/usr/include/onnxruntime
-#cgo linux LDFLAGS: -lonnxruntime -ldl -lstdc++
+#cgo CXXFLAGS: -std=c++17 -I/usr/local/onnxruntime/include
+#cgo linux LDFLAGS: -L/usr/local/onnxruntime/lib -lonnxruntime -ldl -lstdc++
 #cgo darwin LDFLAGS: -lonnxruntime -framework CoreFoundation -lstdc++
 
 #include <stdlib.h>
diff --git a/pkg/opencv/opencv.cpp b/pkg/opencv/opencv.cpp
index 1f6e05a..873f065 100644
--- a/pkg/opencv/opencv.cpp
+++ b/pkg/opencv/opencv.cpp
@@ -1,3 +1,4 @@
+#include <fstream>
 #include <opencv2/opencv.hpp>
 #include <opencv2/dnn.hpp>
 #include <vector>
diff --git a/pkg/opencv/opencv.go b/pkg/opencv/opencv.go
index d6c85e5..9a8d3dd 100644
--- a/pkg/opencv/opencv.go
+++ b/pkg/opencv/opencv.go
@@ -1,9 +1,9 @@
 package opencv
 
 /*
-#cgo CXXFLAGS: -std=c++17 -I/usr/include/opencv4
-#cgo linux LDFLAGS: -L/usr/lib/x86_64-linux-gnu -lopencv_core -lopencv_imgproc -lopencv_imgcodecs -lopencv_dnn -lopencv_calib3d -lstdc++
-#cgo darwin LDFLAGS: -lopencv_core -lopencv_imgproc -lopencv_imgcodecs -lopencv_dnn -lopencv_calib3d -lstdc++
+#cgo CXXFLAGS: -std=c++17 -I/usr/include/opencv4 -I/usr/local/onnxruntime/include
+#cgo linux LDFLAGS: -L/usr/lib/x86_64-linux-gnu -L/usr/local/onnxruntime/lib -lopencv_core -lopencv_imgproc -lopencv_imgcodecs -lopencv_dnn -lopencv_calib3d -lonnxruntime -lstdc++
+#cgo darwin LDFLAGS: -lopencv_core -lopencv_imgproc -lopencv_imgcodecs -lopencv_dnn -lopencv_calib3d -lonnxruntime -lstdc++
 
 #include <stdlib.h>