From a109b264efacf4e0aa9c2dd5a9ef62e371e5419b Mon Sep 17 00:00:00 2001 From: James Date: Wed, 7 Aug 2024 09:48:28 +0100 Subject: [PATCH] fix return type --- supervision/annotators/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py index 133d1e0b..8b672999 100644 --- a/supervision/annotators/core.py +++ b/supervision/annotators/core.py @@ -3,7 +3,7 @@ from typing import List, Optional, Tuple, Union import cv2 import numpy as np -from PIL import ImageDraw, ImageFont +from PIL import ImageDraw, ImageFont, Image from supervision.annotators.base import BaseAnnotator, ImageType from supervision.annotators.utils import ( @@ -2339,4 +2339,4 @@ class BackgroundColorAnnotator(BaseAnnotator): for mask in detections.mask: colored_mask[mask] = scene[mask] - return colored_mask + return Image.fromarray(colored_mask)