diff --git a/test.py b/test.py index 05cc0862..c233adcf 100644 --- a/test.py +++ b/test.py @@ -219,4 +219,4 @@ if __name__ == '__main__': opt.iou_thres, opt.conf_thres, opt.nms_thres, - opt.save_json) + opt.save_json or (opt.data == 'data/coco.data'))