- write error cases to file in test.py

WongKinYiu · Jun 10, 2021 · 31b0ca7 · 31b0ca7
1 parent 8c76b8b
commit 31b0ca7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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