diff --git a/python/examples/detection/detect_image.py b/python/examples/detection/detect_image.py index d2f42c4..3587729 100644 --- a/python/examples/detection/detect_image.py +++ b/python/examples/detection/detect_image.py @@ -94,7 +94,7 @@ def main(): interpreter = make_interpreter(args.model) interpreter.allocate_tensors() - image = Image.open(args.input) + image = Image.open(args.input).convert('RGB') scale = detect.set_input(interpreter, image.size, lambda size: image.resize(size, Image.ANTIALIAS))