From 19cd836c4f2a231c86c4f92a3a0196c36b12fecc Mon Sep 17 00:00:00 2001 From: Jack Kolb Date: Wed, 7 Aug 2024 14:53:55 -0400 Subject: [PATCH] Fixed web demo output value count The demo assumes two return values from GLEEmodel, however the class returns three values. This caused a runtime error. --- app.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/app.py b/app.py index 949a406..eca433e 100644 --- a/app.py +++ b/app.py @@ -166,14 +166,13 @@ def segment_image(img,prompt_mode, categoryname, custom_category, expressiong, r # prompt_list = [mask_ori[0]] prompt_list = [] with torch.no_grad(): - (outputs,_) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=batch_category_name, is_train=False) + (outputs,_,_) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=batch_category_name, is_train=False) topK_instance = max(num_inst_select,1) else: topK_instance = 1 prompt_list = {'grounding':[expressiong]} with torch.no_grad(): - (outputs,_) = GLEEmodel(infer_image, prompt_list, task="grounding", batch_name_list=[], is_train=False) - + (outputs,_,_) = GLEEmodel(infer_image, prompt_list, task="grounding", batch_name_list=[], is_train=False) mask_pred = outputs['pred_masks'][0] mask_cls = outputs['pred_logits'][0] @@ -309,7 +308,7 @@ def segment_image(img,prompt_mode, categoryname, custom_category, expressiong, r prompt_list = {'spatial':[visual_prompt]} with torch.no_grad(): - (outputs,_) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=['object'], is_train=False, visual_prompt_type=prompt_mode ) + (outputs,_,_) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=['object'], is_train=False, visual_prompt_type=prompt_mode ) mask_pred = outputs['pred_masks'][0] mask_cls = outputs['pred_logits'][0]