(image_file)
| 16 | |
| 17 | |
| 18 | def load_image(image_file): |
| 19 | if image_file.startswith('http://') or image_file.startswith('https://'): |
| 20 | response = requests.get(image_file) |
| 21 | image = Image.open(BytesIO(response.content)).convert('RGB') |
| 22 | else: |
| 23 | image = Image.open(image_file).convert('RGB') |
| 24 | return image |
| 25 | |
| 26 | |
| 27 | def main(args): |