Function get_image_shape

caption_anything/utils/utils.py:61–69

(image)

def get_image_shape(image):
    if isinstance(image, str):
        return Image.open(image).size
    elif isinstance(image, np.ndarray):
        return image.shape
    elif isinstance(image, Image.Image):
        return image.size
    else:
        raise NotImplementedError

def is_platform_win():
    return sys.platform == "win32"