Sfoglia il codice sorgente

Merge pull request #48 from Doge2077/master

修复非标准图片base64格式导致图像显示错误
Qing Yan 3 mesi fa
parent
commit
e032665f4c
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      dots_ocr/utils/image_utils.py

+ 1 - 1
dots_ocr/utils/image_utils.py

@@ -68,7 +68,7 @@ def PILimage_to_base64(image, format='PNG'):
     buffered = BytesIO()
     image.save(buffered, format=format)
     base64_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
-    return f"data:image;base64,{base64_str}"
+    return f"data:image/{format.lower()};base64,{base64_str}"
 
 
 def to_rgb(pil_image: Image.Image) -> Image.Image: