From 7dea5e2d31f547d03fc65381e4d38476c5b4dd4f Mon Sep 17 00:00:00 2001 From: Sacha Date: Wed, 10 Apr 2019 11:49:56 +1000 Subject: [PATCH] Increase threshold for NMS The threshold given to NMS (0.3) is far too low and includes many false positives. If someone wanted to see them, they can set the value themselves but I think the default should be slightly more sane. I personally use 0.7 or 0.8. This also improves some benchmarks. Compared to my last patch: PASCAL: 97.28% -> 97.28% (unchanged) AFW: 98.83% -> 99.01% (would be 99.04% with 0.7) --- test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.py b/test.py index 4ba86f5a..daa94cde 100755 --- a/test.py +++ b/test.py @@ -22,7 +22,7 @@ parser.add_argument('--dataset', default='PASCAL', type=str, choices=['AFW', 'PASCAL', 'FDDB'], help='dataset') parser.add_argument('--confidence_threshold', default=0.05, type=float, help='confidence_threshold') parser.add_argument('--top_k', default=5000, type=int, help='top_k') -parser.add_argument('--nms_threshold', default=0.3, type=float, help='nms_threshold') +parser.add_argument('--nms_threshold', default=0.6, type=float, help='nms_threshold') parser.add_argument('--keep_top_k', default=750, type=int, help='keep_top_k') args = parser.parse_args()