deepseekmirror/DeepSeek-VL2
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-VL2.git synced 2025-04-10 21:59:02 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
StevenLiuWen 2025-02-26 13:00:15 +08:00
parent 3c2cd21c9f
commit 699e00ad0d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
deepseek_vl2/serve/app_modules/utils.py
View File

@ -270,7 +270,7 @@ def pil_to_base64(
def parse_ref_bbox(response, image: Image.Image): def parse_ref_bbox(response, image: Image.Image):
try: try:
image = image.copy() image = image.copy()
image_h, image_w = image.size image_w, image_h = image.size
draw = ImageDraw.Draw(image) draw = ImageDraw.Draw(image)
ref = re.findall(r'<\|ref\|>.*?<\|/ref\|>', response) ref = re.findall(r'<\|ref\|>.*?<\|/ref\|>', response)
@ -291,10 +291,10 @@ def parse_ref_bbox(response, image: Image.Image):
for indice, (box, label) in enumerate(zip(boxes, labels)): for indice, (box, label) in enumerate(zip(boxes, labels)):
box = ( box = (
int(box[0] / 999 * image_h), int(box[0] / 999 * image_w),
int(box[1] / 999 * image_w), int(box[1] / 999 * image_h),
int(box[2] / 999 * image_h), int(box[2] / 999 * image_w),
int(box[3] / 999 * image_w), int(box[3] / 999 * image_h),
) )
box_color = BOX2COLOR[indice % len(BOX2COLOR.keys())] box_color = BOX2COLOR[indice % len(BOX2COLOR.keys())]