fix return type

This commit is contained in:
James 2024-08-07 09:48:28 +01:00
parent 389b8968c2
commit a109b264ef
1 changed files with 2 additions and 2 deletions

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)