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()