From 35b5247b5e6d664ae210c57f03b1a4e8d1727853 Mon Sep 17 00:00:00 2001 From: Hamzalopode <46407388+Hamzalopode@users.noreply.github.com> Date: Tue, 28 Mar 2023 13:37:05 +0200 Subject: [PATCH] Documentation typo, and missing value For predict function documentation for postprocess_type argument was incorrect and missing a value. - Fixed GRREDYNMM -> GREEDYNMM. - Added LSNMS for the allowed type. --- sahi/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sahi/predict.py b/sahi/predict.py index 0cca03dd2..00ee4f7de 100644 --- a/sahi/predict.py +++ b/sahi/predict.py @@ -411,7 +411,7 @@ def predict( Default to ``0.2``. postprocess_type: str Type of the postprocess to be used after sliced inference while merging/eliminating predictions. - Options are 'NMM', 'GRREDYNMM' or 'NMS'. Default is 'GRREDYNMM'. + Options are 'NMM', 'GREEDYNMM', 'LSNMS' or 'NMS'. Default is 'GRREDYNMM'. postprocess_match_metric: str Metric to be used during object prediction matching after sliced prediction. 'IOU' for intersection over union, 'IOS' for intersection over smaller area.