Update skip case.

This commit is contained in:
dakotah-jones 2025-01-15 16:41:40 -05:00
parent 830decb483
commit 1ea31bc072

View File

@ -88,6 +88,9 @@ def represent(
# Try load. If load error, will raise exception internal # Try load. If load error, will raise exception internal
img, _ = image_utils.load_image(img_path) img, _ = image_utils.load_image(img_path)
# bgr to rgb
img = img[:, :, ::-1]
if len(img.shape) != 3: if len(img.shape) != 3:
raise ValueError(f"Input img must be 3 dimensional but it is {img.shape}") raise ValueError(f"Input img must be 3 dimensional but it is {img.shape}")