add yolo-pro support

Author: samfreund

photon-client/src/components/settings/ObjectDetectionCard.vue

@@ -227,7 +227,7 @@ const handleBulkImport = () => {
                     :items="
                       useSettingsStore().general.supportedBackends?.includes('RKNN')
                         ? ['YOLOv5', 'YOLOv8', 'YOLO11']
-                        : ['YOLOv8', 'YOLO11']
+                        : ['YOLOv8', 'YOLO11', 'YOLO_PRO']
                     "
                   />
                   <v-btn

photon-client/src/types/SettingTypes.ts

@@ -21,7 +21,7 @@ export interface ObjectDetectionModelProperties {
   resolutionWidth: number;
   resolutionHeight: number;
   family: "RKNN" | "RUBIK";
-  version: "YOLOV5" | "YOLOV8" | "YOLOV11";
+  version: "YOLOV5" | "YOLOV8" | "YOLOV11" | "YOLO_PRO";
 }
 
 export interface MetricData {

photon-core/src/main/java/org/photonvision/common/configuration/NeuralNetworkModelManager.java

@@ -244,7 +244,9 @@ public String extension() {
     public enum Version {
         YOLOV5,
         YOLOV8,
-        YOLOV11
+        YOLOV11,
+        YOLOV11_OBB, // Not yet supported
+        YOLO_PRO
     }
 
     /**

photon-core/src/main/java/org/photonvision/vision/objects/RubikModel.java

@@ -51,7 +51,9 @@ public RubikModel(ModelProperties properties) throws IllegalArgumentException {
             throw new IllegalArgumentException("Model family must be RUBIK");
         }
 
-        if (properties.version() != Version.YOLOV8 && properties.version() != Version.YOLOV11) {
+        if (properties.version() != Version.YOLOV8
+                && properties.version() != Version.YOLOV11
+                && properties.version() != Version.YOLO_PRO) {
             throw new IllegalArgumentException("Model version must be YOLOV8 or YOLOV11");
         }
 
@@ -77,7 +79,9 @@ public ModelProperties getProperties() {
 
     public ObjectDetector load() {
         return new RubikObjectDetector(
-                this, new Size(this.properties.resolutionWidth(), this.properties.resolutionHeight()));
+                this,
+                new Size(this.properties.resolutionWidth(), this.properties.resolutionHeight()),
+                this.properties.version().ordinal());
     }
 
     public String toString() {

photon-server/src/main/java/org/photonvision/server/RequestHandler.java

@@ -571,6 +571,7 @@ public static void onImportObjectDetectionModelRequest(Context ctx) {
                         case "YOLOv5" -> NeuralNetworkModelManager.Version.YOLOV5;
                         case "YOLOv8" -> NeuralNetworkModelManager.Version.YOLOV8;
                         case "YOLO11" -> NeuralNetworkModelManager.Version.YOLOV11;
+                        case "YOLO_PRO" -> NeuralNetworkModelManager.Version.YOLO_PRO;
                             // Add more versions as necessary for new models
                         default -> {
                             ctx.status(400);