update test.py

This commit is contained in:
Glenn Jocher committed 2020-07-15 19:06:39 -07:00
1 parent 6bd9218922
commit 1ce686edfc
1 file changed
+1 -1
+1 -1
View File
@@ -236,7 +236,7 @@ if __name__ == '__main__':
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)