diff --git a/test.py b/test.py index 8e0df4e..a9231f1 100644 --- a/test.py +++ b/test.py @@ -312,6 +312,7 @@ def test(data, parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment') parser.add_argument('--cfg', type=str, default='models/yolov4-csp.cfg', help='*.cfg path') parser.add_argument('--names', type=str, default='data/coco.names', help='*.cfg path') + parser.add_argument('--save-errors', help='Save the error cases to file', action='store_true') opt = parser.parse_args() opt.save_json |= opt.data.endswith('coco.yaml') opt.data = check_file(opt.data) # check file