Uncomment save_json

neuralmagic · Jan 24, 2023 · 4b6f655 · 4b6f655
1 parent 85c4bf9
commit 4b6f655
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/segment/val.py b/segment/val.py
@@ -426,7 +426,7 @@ def parse_opt():
     parser.add_argument('--dnn', action='store_true', help='use OpenCV DNN for ONNX inference')
     opt = parser.parse_args()
     opt.data = check_yaml(opt.data)  # check YAML
-    # opt.save_json |= opt.data.endswith('coco.yaml')
+    opt.save_json |= opt.data.endswith('coco.yaml')
     opt.save_txt |= opt.save_hybrid
     print_args(vars(opt))
     return opt