bugfix

src/common/data.hpp

@@ -5,7 +5,7 @@
 #include <vector>
 #include <map>
 #include <iostream>
-
+#include <memory>
 #include "common/check.hpp"
 
 namespace data

src/infer/infer.cpp

@@ -12,8 +12,9 @@ std::shared_ptr<Infer> load(const std::string& model_path, ModelType model_type,
     switch (model_type)
     {
     case ModelType::YOLOV5:
-    case ModelType::YOLOV8:
+    case ModelType::YOLOV5SEG:
     case ModelType::YOLO11:
+    case ModelType::YOLO11SEG:
     case ModelType::YOLO11POSE:
         infer = yolo::load_yolo(model_path, model_type, names, gpu_id, confidence_threshold, nms_threshold);
         break;