quoccuongle/grounding-dino
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

eval empty token_spans properly (#191)

Browse Source
This commit is contained in:
Kazuto Murase
2023-11-13 14:53:12 +09:00
committed by GitHub
parent eeba084341
commit fbb2532bb0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
demo/inference_on_a_image.py
View File

@@ -199,7 +199,7 @@ if __name__ == "__main__":
# run model # run model
boxes_filt, pred_phrases = get_grounding_output( boxes_filt, pred_phrases = get_grounding_output(
model, image, text_prompt, box_threshold, text_threshold, cpu_only=args.cpu_only, token_spans=eval(token_spans) model, image, text_prompt, box_threshold, text_threshold, cpu_only=args.cpu_only, token_spans=eval(f"{token_spans}")
) )
# visualize pred # visualize pred