fix main.cpp

Author: MatthijsBurgh

src/main.cpp

@@ -13,7 +13,7 @@ int main()
     std::tie(yoloDetector, params) = Initialize();
 
     std::filesystem::path current_path = std::filesystem::current_path();
-    std::filesystem::path imgs_path = "/home/amigo/Documents/repos/hero_sam.bak/sam_inference/build/images";
+    std::filesystem::path imgs_path = current_path / "images";
     for (auto& i : std::filesystem::directory_iterator(imgs_path))
     {
         if (i.path().extension() == ".jpg" || i.path().extension() == ".png" || i.path().extension() == ".jpeg")
@@ -22,6 +22,12 @@ int main()
             cv::Mat img = cv::imread(img_path);
             std::vector<DL_RESULT> results;
             results = Detector(yoloDetector, img);
+            for (const auto& result : results)
+            {
+                std::cout << "Image path: " << img_path << "\n"
+                          << "class id:   " << result.classId << "\n"
+                          << "confidence: " << result.confidence << "\n";
+            }
         }
     }