Commit e2f6174b authored by Jonathan Passerat-Palmbach's avatar Jonathan Passerat-Palmbach
Browse files

fix: force conversion to RGB to solve runtime error

parent 836ea750
......@@ -16,4 +16,6 @@ image = load(parsed.inputfile).get_data()
image = Image.fromarray(image[:, :, image.shape[2]//2] if image.ndim == 3
else image)
image = image.resize([128, 128])
if image.mode != 'RGB':
image = image.convert('RGB')
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