浏览代码

Fix:Repeated images when parsing markdown files with images (#9196)

### What problem does this PR solve?

https://github.com/infiniflow/ragflow/issues/9149

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.20.1
Stephen Hu 2 个月前
父节点
当前提交
667c5812d0
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 13 次插入1 次删除
  1. 13
    1
      rag/nlp/__init__.py

+ 13
- 1
rag/nlp/__init__.py 查看文件

@@ -634,6 +634,17 @@ def concat_img(img1, img2):
return img2
if not img1 and not img2:
return None
if img1 is img2:
return img1
if isinstance(img1, Image.Image) and isinstance(img2, Image.Image):
pixel_data1 = img1.tobytes()
pixel_data2 = img2.tobytes()
if pixel_data1 == pixel_data2:
img2.close()
return img1

width1, height1 = img1.size
width2, height2 = img2.size

@@ -643,7 +654,8 @@ def concat_img(img1, img2):

new_image.paste(img1, (0, 0))
new_image.paste(img2, (0, height1))

img1.close()
img2.close()
return new_image



正在加载...
取消
保存