fumucu commited on
Commit
74d88b9
·
1 Parent(s): 2a2872c

initial commit

Browse files
Files changed (1) hide show
  1. laod_pipeline.py +1 -1
laod_pipeline.py CHANGED
@@ -73,7 +73,7 @@ def laod_gdino(image):
73
  results = processor.post_process_grounded_object_detection(
74
  outputs,
75
  inputs.input_ids,
76
- box_threshold=0.4,
77
  text_threshold=0.3,
78
  target_sizes=[image.size[::-1]]
79
  )
 
73
  results = processor.post_process_grounded_object_detection(
74
  outputs,
75
  inputs.input_ids,
76
+ threshold=0.4,
77
  text_threshold=0.3,
78
  target_sizes=[image.size[::-1]]
79
  )