update test.py

fhkiel-mlaip · Jul 16, 2020 · 45dd81c · 45dd81c
1 parent 62c106d
commit 45dd81c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -236,7 +236,7 @@ def test(data,
     parser.add_argument('--merge', action='store_true', help='use Merge NMS')
     parser.add_argument('--verbose', action='store_true', help='report mAP by class')
     opt = parser.parse_args()
-    opt.save_json = opt.save_json or opt.data.endswith('coco.yaml')
+    opt.save_json |= opt.data.endswith('coco.yaml')
     opt.data = check_file(opt.data)  # check file
     print(opt)