Update model references from YOLO11 to YOLO26 across configuration and schema files

Author: yihong1120

examples/YOLO_server_api/backend/config.py

@@ -13,13 +13,13 @@
 
 # Base model variants
 MODEL_VARIANTS_ENV: str = os.getenv(
-    'MODEL_VARIANTS', 'yolo11x,yolo11l,yolo11m,yolo11s,yolo11n',
+    'MODEL_VARIANTS', 'yolo26x,yolo26l,yolo26m,yolo26s,yolo26n',
 )
 MODEL_VARIANTS: list[str] = [
     v.strip() for v in MODEL_VARIANTS_ENV.split(',') if v.strip()
 ]
 if not MODEL_VARIANTS:  # Fallback protection
-    MODEL_VARIANTS = ['yolo11n']
+    MODEL_VARIANTS = ['yolo26n']
 
 # Whether to enable lazy loading of models:
 # True means models are loaded only when first used

examples/YOLO_server_api/backend/schemas.py

@@ -26,7 +26,7 @@ class ModelFileUpdate(BaseModel):
     Represents the data required to update a model file.
 
     Attributes:
-        model (str): The model identifier (e.g., 'yolo11n').
+        model (str): The model identifier (e.g., 'yolo26n').
         file (UploadFile): The uploaded file (e.g., .pt file).
     """
 

src/live_stream_detection.py

@@ -39,7 +39,7 @@ class LiveStreamDetector:
     def __init__(
         self,
         api_url: str | None = None,
-        model_key: str = 'yolo11n',
+        model_key: str = 'yolo26n',
         output_folder: str | None = None,
         detect_with_server: bool = False,
         shared_token: SharedToken | None = None,
@@ -105,7 +105,7 @@ def __init__(
                 )
             else:
                 self.model = AutoDetectionModel.from_pretrained(
-                    'yolo11',
+                    'yolo26',
                     model_path=str(
                         Path('models/pt') /
                         f"best_{self.model_key}.pt",
@@ -1236,7 +1236,7 @@ async def main() -> None:
     )
     parser.add_argument(
         '--model_key', type=str,
-        default='yolo11n', help='YOLO model identifier key',
+        default='yolo26n', help='YOLO model identifier key',
     )
     parser.add_argument(
         '--detect_with_server',

src/live_stream_tracker.py

@@ -19,20 +19,20 @@ class DetectionResult(TypedDict):
 
 class LiveStreamDetector:
     """
-    A class to perform live stream detection and tracking using YOLO11.
+    A class to perform live stream detection and tracking using YOLO26.
     """
 
     def __init__(
         self,
         stream_url: str,
-        model_path: str = '../models/pt/best_yolo11n.pt',
+        model_path: str = '../models/pt/best_yolo26n.pt',
     ):
         """
         Initialise live stream detector with video URL, YOLO model path.
 
         Args:
             stream_url (str): The full URL to the live video stream.
-            model_path (str): The path to the YOLO11 model file.
+            model_path (str): The path to the YOLO26 model file.
         """
         self.stream_url = stream_url
         self.model_path = model_path
@@ -60,7 +60,7 @@ def generate_detections(
             now = datetime.datetime.now()
             timestamp = now.timestamp()  # Unix timestamp
 
-            # Run YOLO11 tracking, maintain tracks across frames
+            # Run YOLO26 tracking, maintain tracks across frames
             results = self.model.track(source=frame, persist=True)
 
             # If detections exist, extract IDs and data
@@ -117,7 +117,7 @@ def main():
     Main function to run the live stream detection.
     """
     parser = argparse.ArgumentParser(
-        description='Perform live stream detection and tracking using YOLO11.',
+        description='Perform live stream detection and tracking using YOLO26.',
     )
     parser.add_argument(
         '--url',
@@ -128,7 +128,7 @@ def main():
     parser.add_argument(
         '--model',
         type=str,
-        default='../models/yolo11n.pt',
+        default='../models/pt/yolo26n.pt',
         help='Path to the YOLO model',
     )
     args = parser.parse_args()

src/model_fetcher.py

@@ -35,7 +35,7 @@ def __init__(
         """
         self.api_url = api_url
         self.models = models or [
-            'yolo11n', 'yolo11s', 'yolo11m', 'yolo11l', 'yolo11x',
+            'yolo26n', 'yolo26s', 'yolo26m', 'yolo26l', 'yolo26x',
         ]
         self.local_dir = Path(local_dir)