fix doc

Koldim2001 · Sep 23, 2024 · d6740f2 · d6740f2
1 parent 53a455f
commit d6740f2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -162,7 +162,7 @@ Possible arguments of the ```visualize_results``` function:
 | show_class              | bool            | True          | Whether to show class labels.                                                                 |
 | fill_mask               | bool            | False         | Whether to fill the segmented regions with color.                                             |
 | alpha                   | float           | 0.3           | The transparency of filled masks.                                                             |
-| color_class_background  | tuple           | (0, 0, 255)   | The background BGR color for class labels.                                                    |
+| color_class_background  | tuple / list          | (0, 0, 255)   | The background BGR color for class labels. If you pass a list of tuples, then each class will have its own color.                                                  |
 | color_class_text        | tuple           |(255, 255, 255)| The text color for class labels.                                                              |
 | thickness               | int             | 4             | The thickness of bounding box and text.                                                       |
 | font                    | cv2.font        |cv2.FONT_HERSHEY_SIMPLEX | The font type for class labels.                                                     |

diff --git a/patched_yolo_infer/README.md b/patched_yolo_infer/README.md
@@ -130,7 +130,7 @@ Visualizes custom results of object detection or segmentation on an image.
 - **show_class** (*bool*): Whether to show class labels. Default is True.
 - **fill_mask** (*bool*): Whether to fill the segmented regions with color. Default is False.
 - **alpha** (*float*): The transparency of filled masks. Default is 0.3.
-- **color_class_background** (*tuple*): The background BGR color for class labels. Default is (0, 0, 255) (red).
+- **color_class_background** (*tuple/list*): The background BGR color for class labels. If you pass a list of tuples, then each class will have its own color. Default is (0, 0, 255) (red).
 - **color_class_text** (*tuple*): The text color for class labels. Default is (255, 255, 255) (white).
 - **thickness** (*int*): The thickness of bounding box and text. Default is 4.
 - **font**: The font type for class labels. Default is cv2.FONT_HERSHEY_SIMPLEX.