Skip to content

Commit dc938d8

Browse files
authored
Update docstring for greedy_nmm (#1205)
1 parent 05e4019 commit dc938d8

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

sahi/postprocess/combine.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -189,13 +189,10 @@ def greedy_nmm(
189189
Args:
190190
object_predictions_as_tensor: (tensor) The location preds for the image
191191
along with the class predscores, Shape: [num_boxes,5].
192-
object_predictions_as_list: ObjectPredictionList Object prediction objects
193-
to be merged.
194192
match_metric: (str) IOU or IOS
195-
match_threshold: (float) The overlap thresh for
196-
match metric.
193+
match_threshold: (float) The overlap thresh for match metric.
197194
Returns:
198-
keep_to_merge_list: (Dict[int:List[int]]) mapping from prediction indices
195+
keep_to_merge_list: (dict[int, list[int]]) mapping from prediction indices
199196
to keep to a list of prediction indices to be merged.
200197
"""
201198
keep_to_merge_list = {}

0 commit comments

Comments
 (0)