refactor: Update default models to yolo11n and yoloe-11s in training and deployment scripts

Author: Jordan-Pierce

coralnet_toolbox/MVAT/core/Frustum.py

@@ -1,5 +1,6 @@
 import numpy as np
 import pyvista as pv
+
 from PyQt5.QtGui import QImage
 
 # ----------------------------------------------------------------------------------------------------------------------

coralnet_toolbox/MachineLearning/TrainModel/QtClassify.py

@@ -82,4 +82,4 @@ def load_model_combobox(self):
             self.model_combo.addItems(list(community_configs.keys()))
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yolo26n-cls.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yolo11n-cls.pt'))

coralnet_toolbox/MachineLearning/TrainModel/QtDetect.py

@@ -117,4 +117,4 @@ def load_model_combobox(self):
             self.model_combo.addItems(list(community_configs.keys()))
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yolo26n.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yolo11n.pt'))

coralnet_toolbox/MachineLearning/TrainModel/QtSegment.py

@@ -85,4 +85,4 @@ def load_model_combobox(self):
             self.model_combo.addItems(list(community_configs.keys()))
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yolo26n-seg.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yolo11n-seg.pt'))

coralnet_toolbox/MachineLearning/TuneModel/QtClassify.py

@@ -72,4 +72,4 @@ def load_model_combobox(self):
             self.model_combo.addItems(list(community_configs.keys()))
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yolo26n-cls.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yolo11n-cls.pt'))

coralnet_toolbox/MachineLearning/TuneModel/QtDetect.py

@@ -107,4 +107,4 @@ def load_model_combobox(self):
             self.model_combo.addItems(list(community_configs.keys()))
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yolo26n.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yolo11n.pt'))

coralnet_toolbox/MachineLearning/TuneModel/QtSegment.py

@@ -75,4 +75,4 @@ def load_model_combobox(self):
             self.model_combo.addItems(list(community_configs.keys()))
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yolo26n-seg.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yolo11n-seg.pt'))

coralnet_toolbox/SeeAnything/QtDeployGenerator.py

@@ -385,7 +385,7 @@ def setup_models_layout(self):
             self.model_combo.addItem(model_name)
 
         # Set the default model
-        self.model_combo.setCurrentIndex(self.models.index('yoloe-26s-seg.pt'))
+        self.model_combo.setCurrentIndex(self.models.index('yoloe-11s-seg.pt'))
         model_select_layout.addRow("Model:", self.model_combo)
 
         # Add VPE file selection to the first tab

coralnet_toolbox/SeeAnything/QtDeployPredictor.py

@@ -140,7 +140,7 @@ def setup_models_layout(self):
         self.model_combo.addItems(standard_models)
 
         # Set the default model
-        self.model_combo.setCurrentIndex(standard_models.index('yoloe-26s-seg.pt'))
+        self.model_combo.setCurrentIndex(standard_models.index('yoloe-11s-seg.pt'))
         model_select_layout.addRow("Model:", self.model_combo)
 
         tab_widget.addTab(model_select_tab, "Select Model")

coralnet_toolbox/SeeAnything/QtTrainModel.py

@@ -714,7 +714,7 @@ def load_model_combobox(self):
                 'yoloe-26l-seg.pt',
                 'yoloe-26x-seg.pt'
             ]
-            default_model = 'yoloe-26s-seg.pt'
+            default_model = 'yoloe-11s-seg.pt'
         else:  # detect
             standard_models = [
                 'yoloe-v8s.yaml',
@@ -729,7 +729,7 @@ def load_model_combobox(self):
                 'yoloe-26l.yaml',
                 'yoloe-26x.yaml'
             ]
-            default_model = 'yoloe-26s.yaml'
+            default_model = 'yoloe-11s.yaml'
 
         self.model_combo.addItems(standard_models)