fix return type

This commit is contained in:
James committed 2024-08-07 09:48:28 +01:00
1 parent 389b8968c2
commit a109b264ef
1 file changed
+2 -2
+2 -2
View File
@@ -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)