From 86442fc40e50c0392e1d3e8308781cdee3f11e4a Mon Sep 17 00:00:00 2001 From: Smit Lunagariya Date: Thu, 14 Mar 2024 05:58:25 +0000 Subject: [PATCH] Fix docstring of pyramid keys --- keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py b/keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py index 6c17c71a72..94553ef106 100644 --- a/keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py +++ b/keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py @@ -331,7 +331,7 @@ class YOLOV8Detector(Task): Args: backbone: `keras.Model`, must implement the `pyramid_level_inputs` - property with keys "P2", "P3", and "P4" and layer names as values. + property with keys "P3", "P4", and "P5" and layer names as values. A sensible backbone to use is the `keras_cv.models.YOLOV8Backbone`. num_classes: integer, the number of classes in your dataset excluding the background class. Classes should be represented by integers in the