fixed functionallity for the tests to pass and added logging definition

Author: IasonTheodorou

src/main.cpp

@@ -3,6 +3,7 @@
 #include <vector>
 #include <filesystem>
 #include <opencv2/opencv.hpp>
+
 int main()
 {
     // Running inference
@@ -11,7 +12,7 @@ int main()
     SEG::DL_INIT_PARAM params_decoder;
     std::tie(samSegmentors, params_encoder, params_decoder) = Initializer();
     std::filesystem::path current_path = std::filesystem::current_path();
-    std::filesystem::path imgs_path =  "/home/amigo/Documents/repos/hero_sam/pipeline/build/images"; // current_path / <- you could use
+    std::filesystem::path imgs_path =  "/home/amigo/Documents/repos/hero_sam/sam_inference/build/images"; // current_path / <- you could use
     for (auto &i : std::filesystem::directory_iterator(imgs_path))
     {
         if (i.path().extension() == ".jpg" || i.path().extension() == ".png" || i.path().extension() == ".jpeg")

src/sam_inference.cpp

@@ -3,7 +3,7 @@
 #include <regex>
 
 #define benchmark
-#define ROI
+//#define ROI
 
 SAM::SAM() {}
 
@@ -223,6 +223,8 @@ const char *SAM::TensorProcess(clock_t &starttime_1, const cv::Mat &iImg,
 
     std::vector<cv::Rect> boundingBoxes;
     boundingBoxes.push_back(bbox);
+#else
+    result.boxes.push_back(cv::Rect(0, 0, iImg.cols, iImg.rows));
 #endif // ROI
 
 #ifdef benchmark

src/utils.cpp

@@ -1,5 +1,6 @@
 #include "utils.h"
 #include <opencv2/ximgproc/edge_filter.hpp> // for guided filter
+#define LOGGING
 
 // Constructor
 Utils::Utils()
@@ -33,7 +34,8 @@ char *Utils::PreProcess(const cv::Mat &iImg, std::vector<int> iImgSize, cv::Mat
         resizeScales = iImg.rows / (float)iImgSize.at(1);
         cv::resize(oImg, oImg, cv::Size(int(iImg.cols / resizeScales), iImgSize.at(1)));
     }
-    cv::Mat tempImg = cv::Mat::zeros(iImgSize.at(0), iImgSize.at(1), CV_8UC3);
+    //cv::Mat tempImg = cv::Mat::zeros(iImgSize.at(0), iImgSize.at(1), CV_8UC3);
+    cv::Mat tempImg = cv::Mat::zeros(iImgSize.at(1), iImgSize.at(0), CV_8UC3);
     oImg.copyTo(tempImg(cv::Rect(0, 0, oImg.cols, oImg.rows)));
     oImg = tempImg;
 
@@ -198,6 +200,7 @@ void Utils::PostProcess(std::vector<Ort::Value> &output_tensors, const cv::Mat &
         result.masks.push_back(finalMask);
 
         // Overlay for display on a copy (iImg is const)
+        #ifdef LOGGING
         cv::Mat overlay = iImg.clone();
         std::vector<std::vector<cv::Point>> contours;
         cv::findContours(finalMask, contours, cv::RETR_EXTERNAL, cv::CHAIN_APPROX_SIMPLE);
@@ -210,6 +213,7 @@ void Utils::PostProcess(std::vector<Ort::Value> &output_tensors, const cv::Mat &
         cv::imshow("SAM Segmentation", overlay);
         cv::waitKey(0);
         cv::destroyAllWindows();
+        #endif // LOGGING
     }
     else
     {