Skip to content

GitLab

Commit 74507ce8 authored by Geewook Kim's avatar Geewook Kim
Browse files

fix: ensure RGB

parent 0e9a705a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
donut/model.py
View file @ 74507ce8
...@@ -108,6 +108,7 @@ class SwinEncoder(nn.Module): ...@@ -108,6 +108,7 @@ class SwinEncoder(nn.Module):
- rotate (if align_long_axis is True and image is not aligned longer axis with canvas) - rotate (if align_long_axis is True and image is not aligned longer axis with canvas)
- pad - pad
""" """
img = img.convert("RGB")
if self.align_long_axis and ( if self.align_long_axis and (
(self.input_size[0] > self.input_size[1] and img.width > img.height) (self.input_size[0] > self.input_size[1] and img.width > img.height)
or (self.input_size[0] < self.input_size[1] and img.width < img.height) or (self.input_size[0] < self.input_size[1] and img.width < img.height)
......
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