There was an error while loading. Please reload this page.
1 parent 05e4019 commit dc938d8Copy full SHA for dc938d8
1 file changed
sahi/postprocess/combine.py
@@ -189,13 +189,10 @@ def greedy_nmm(
189
Args:
190
object_predictions_as_tensor: (tensor) The location preds for the image
191
along with the class predscores, Shape: [num_boxes,5].
192
- object_predictions_as_list: ObjectPredictionList Object prediction objects
193
- to be merged.
194
match_metric: (str) IOU or IOS
195
- match_threshold: (float) The overlap thresh for
196
- match metric.
+ match_threshold: (float) The overlap thresh for match metric.
197
Returns:
198
- keep_to_merge_list: (Dict[int:List[int]]) mapping from prediction indices
+ keep_to_merge_list: (dict[int, list[int]]) mapping from prediction indices
199
to keep to a list of prediction indices to be merged.
200
"""
201
keep_to_merge_list = {}
0 commit comments