ML: throw errors when fail to load .onnx or _label.txt

Author: jw098

SerialPrograms/Source/ML/Inference/ML_YOLOv5Detector.cpp

@@ -10,6 +10,7 @@
 #include <QMessageBox>
 #include <opencv2/imgproc.hpp>
 #include <opencv2/imgcodecs.hpp>
+#include "Common/Cpp/Exceptions.h"
 #include "CommonFramework/ImageTypes/ImageViewRGB32.h"
 #include "CommonFramework/VideoPipeline/VideoOverlay.h"
 #include "CommonFramework/VideoPipeline/VideoOverlayScopes.h"
@@ -32,29 +33,38 @@ YOLOv5Detector::~YOLOv5Detector() = default;
 YOLOv5Detector::YOLOv5Detector(const std::string& model_path)
 {
     if (!model_path.ends_with(".onnx")){
-        std::cerr << "Error: wrong model path extension: " << model_path << ". It must be .onnx" << std::endl;
-        QMessageBox box;
-        box.critical(nullptr, "Wrong Model Extension",
-            QString::fromStdString("YOLOv5 model path must end with .onnx. But got " + model_path + "."));
-        return;
+        throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, 
+            "Error: YOLOv5 model path must end with .onnx. But got " + model_path + ".");
+
+        // std::cerr << "Error: wrong model path extension: " << model_path << ". It must be .onnx" << std::endl;
+        // QMessageBox box;
+        // box.critical(nullptr, "Wrong Model Extension",
+        //     QString::fromStdString("YOLOv5 model path must end with .onnx. But got " + model_path + "."));
+        // return;
     }
 
     std::string label_file_path = model_path.substr(0, model_path.size() - 5) + "_label.txt";
     std::vector<std::string> labels;
     if (!std::filesystem::exists(label_file_path)){
-        std::cerr << "Error: no such YOLOv5 label file path " << label_file_path << "." << std::endl;
-        QMessageBox box;
-        box.critical(nullptr, "YOLOv5 Label File Does Not Exist",
-            QString::fromStdString("YOLOv5 label file path " + label_file_path + " does not exist."));
-        return;
+        throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, 
+            "Error: YOLOv5 label file path " + label_file_path + " does not exist.");
+
+        // std::cerr << "Error: no such YOLOv5 label file path " << label_file_path << "." << std::endl;
+        // QMessageBox box;
+        // box.critical(nullptr, "YOLOv5 Label File Does Not Exist",
+        //     QString::fromStdString("YOLOv5 label file path " + label_file_path + " does not exist."));
+        // return;
     }
     std::ifstream label_file(label_file_path);
     if (!label_file.is_open()){
-        std::cerr << "Error: failed to open YOLOv5 label file " << label_file_path << "." << std::endl;
-        QMessageBox box;
-        box.critical(nullptr, "Cannot Open YOLOv5 Label File",
-            QString::fromStdString("YOLOv5 label file " + label_file_path + " cannot be opened."));
-        return;
+        throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, 
+            "Error: YOLOv5 label file " + label_file_path + " cannot be opened.");
+
+        // std::cerr << "Error: failed to open YOLOv5 label file " << label_file_path << "." << std::endl;
+        // QMessageBox box;
+        // box.critical(nullptr, "Cannot Open YOLOv5 Label File",
+        //     QString::fromStdString("YOLOv5 label file " + label_file_path + " cannot be opened."));
+        // return;
     }
     std::string line;
     while (std::getline(label_file, line)){