归并

pp.py

@@ -30,7 +30,7 @@ class MyEncoder(json.JSONEncoder):
         return super(MyEncoder, self).default(obj)
 
 
-def download_image(url: str) -> Image.Image:
+def download_image(url: str, max_size=32767) -> Image.Image:
     if url.endswith('.gif') or url.endswith('.GIF'):
         print(f'跳过GIF {url}')
         return None
@@ -44,6 +44,9 @@ def download_image(url: str) -> Image.Image:
             img = Image.open(io.BytesIO(response.content))
         if img.mode != 'RGB':
             img = img.convert('RGB')
+        if img.size[0] > max_size or img.size[1] > max_size:
+            print(f'跳过尺寸过大的图像 {url}')
+            return None
         return img
     except Exception as e:
         print(f'图片从{url}下载失败，错误信息为：{e}')