Commit ae8f23b9 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add P check

parent b5ee0509

imgutils/data/image.py

+1 −1
Original line number Diff line number Diff line
@@ -19,7 +19,7 @@ MultiImagesTyping = Union[ImageTyping, List[ImageTyping], Tuple[ImageTyping, ... 




def _has_alpha_channel(image: Image.Image) -> bool:

    return any(band in {'A', 'a'} for band in image.getbands())

    return any(band in {'A', 'a', 'P'} for band in image.getbands())





def load_image(image: ImageTyping, mode=None, force_background: Optional[str] = 'white'):

imgutils/detect/text.py

+1 −1
Original line number Diff line number Diff line
@@ -142,7 +142,7 @@ def detect_text(image: ImageTyping, model: str = _DEFAULT_MODEL, threshold: floa 
        This function is deprecated, and it will be removed from imgutils in the future.

        Please migrate to :func:`imgutils.ocr.detect_text_with_ocr` as soon as possible.

    """

    image = load_image(image)

    image = load_image(image, mode='RGB')

    if max_area_size is not None and image.width * image.height >= max_area_size ** 2:

        r = ((image.width * image.height) / (max_area_size ** 2)) ** 0.5

        new_width, new_height = int(image.width / r), int(image.height / r)