"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "2735e9e3c9bd68540e6e337a6379a2a35bd2c9f9"
Commit b3b5ddb0 authored by comfyanonymous's avatar comfyanonymous
Browse files

Support I mode images in LoadImageMask.

parent 2d74fc43
...@@ -1472,6 +1472,8 @@ class LoadImageMask: ...@@ -1472,6 +1472,8 @@ class LoadImageMask:
i = Image.open(image_path) i = Image.open(image_path)
i = ImageOps.exif_transpose(i) i = ImageOps.exif_transpose(i)
if i.getbands() != ("R", "G", "B", "A"): if i.getbands() != ("R", "G", "B", "A"):
if i.mode == 'I':
i = i.point(lambda i: i * (1 / 255))
i = i.convert("RGBA") i = i.convert("RGBA")
mask = None mask = None
c = channel[0].upper() c = channel[0].upper()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment