From a3304818d2bfc78d16a8ffa51f0589a42a4b96af Mon Sep 17 00:00:00 2001 From: Sam Freund Date: Sun, 9 Mar 2025 17:11:56 -0500 Subject: [PATCH] fix: docs for YOLOv11 naming (#1806) --- docs/source/docs/objectDetection/about-object-detection.md | 4 ++-- .../src/components/settings/ObjectDetectionCard.vue | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docs/source/docs/objectDetection/about-object-detection.md b/docs/source/docs/objectDetection/about-object-detection.md index c6bbe36f9..40ee828ae 100644 --- a/docs/source/docs/objectDetection/about-object-detection.md +++ b/docs/source/docs/objectDetection/about-object-detection.md @@ -44,11 +44,11 @@ Before beginning, it is necessary to install the [rknn-toolkit2](https://github. ## Uploading Custom Models :::{warning} -PhotonVision currently ONLY supports 640x640 Ultralytics YOLOv5, YOLOv8, and YOLO11 models trained and converted to `.rknn` format for RK3588 CPUs! Other models require different post-processing code and will NOT work. The model conversion process is also highly particular. Proceed with care. +PhotonVision currently ONLY supports 640x640 Ultralytics YOLOv5, YOLOv8, and YOLOv11 models trained and converted to `.rknn` format for RK3588 CPUs! Other models require different post-processing code and will NOT work. The model conversion process is also highly particular. Proceed with care. ::: In the settings, under `Device Control`, there's an option to upload a new object detection model. Naming convention -should be `name-verticalResolution-horizontalResolution-modelType`. The +should be `name-verticalResolution-horizontalResolution-yolovXXX`. The `name` should only include alphanumeric characters, periods, and underscores. Additionally, the labels file ought to have the same name as the RKNN file, with `-labels` appended to the end. For example, if the RKNN file is named `Algae_1.03.2025-640-640-yolov5s.rknn`, the labels file should be diff --git a/photon-client/src/components/settings/ObjectDetectionCard.vue b/photon-client/src/components/settings/ObjectDetectionCard.vue index 2e3bcd9de..085ce378a 100644 --- a/photon-client/src/components/settings/ObjectDetectionCard.vue +++ b/photon-client/src/components/settings/ObjectDetectionCard.vue @@ -88,12 +88,12 @@ const supportedModels = computed(() => { Import New Object Detection Model Upload a new object detection model to this device that can be used in a pipeline. Naming convention - should be name-verticalResolution-horizontalResolution-modelType. The + should be name-verticalResolution-horizontalResolution-yolovXXX. The name should only include alphanumeric characters, periods, and underscores. Additionally, the labels file ought to have the same name as the RKNN file, with -labels appended to the end. For example, if the RKNN file is named note-640-640-yolov5s.rknn, the labels file - should be named note-640-640-yolov5s-labels.txt. Note that ONLY 640x640 YOLOv5 & YOLOv8 - models trained and converted to `.rknn` format for RK3588 CPUs are currently supported! + should be named note-640-640-yolov5s-labels.txt. Note that ONLY 640x640 YOLOv5, YOLOv8, and + YOLOv11 models trained and converted to `.rknn` format for RK3588 CPUs are currently supported!