Skip to content

Commit 707132b

Browse files
authored
Fix docstring of pyramid keys (#2391)
1 parent 1b4e2ab commit 707132b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_cv/models/object_detection/yolo_v8/yolo_v8_detector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ class YOLOV8Detector(Task):
331331
332332
Args:
333333
backbone: `keras.Model`, must implement the `pyramid_level_inputs`
334-
property with keys "P2", "P3", and "P4" and layer names as values.
334+
property with keys "P3", "P4", and "P5" and layer names as values.
335335
A sensible backbone to use is the `keras_cv.models.YOLOV8Backbone`.
336336
num_classes: integer, the number of classes in your dataset excluding the
337337
background class. Classes should be represented by integers in the

0 commit comments

Comments
 (0)