glenn-jocher
commited on
Commit
•
1ce686e
1
Parent(s):
6bd9218
update test.py
Browse files
test.py
CHANGED
@@ -236,7 +236,7 @@ if __name__ == '__main__':
|
|
236 |
parser.add_argument('--merge', action='store_true', help='use Merge NMS')
|
237 |
parser.add_argument('--verbose', action='store_true', help='report mAP by class')
|
238 |
opt = parser.parse_args()
|
239 |
-
opt.save_json
|
240 |
opt.data = check_file(opt.data) # check file
|
241 |
print(opt)
|
242 |
|
|
|
236 |
parser.add_argument('--merge', action='store_true', help='use Merge NMS')
|
237 |
parser.add_argument('--verbose', action='store_true', help='report mAP by class')
|
238 |
opt = parser.parse_args()
|
239 |
+
opt.save_json |= opt.data.endswith('coco.yaml')
|
240 |
opt.data = check_file(opt.data) # check file
|
241 |
print(opt)
|
242 |
|