diff --git a/src/inference.py b/src/inference.py index 564dd7685..65ee6a0c1 100644 --- a/src/inference.py +++ b/src/inference.py @@ -884,7 +884,7 @@ def main(): for count, img_file in enumerate(img_files): print("({}/{})".format(count+1, num_files)) img_path = os.path.join(args.image_dir, img_file) - img = Image.open(img_path) + img = Image.open(img_path).convert('RGB') print("Image loaded.") if not args.words_dir is None: