glenn-jocher
commited on
Commit
·
208493d
1
Parent(s):
948bcdd
updates
Browse files
test.py
CHANGED
@@ -247,7 +247,7 @@ if __name__ == '__main__':
|
|
247 |
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
248 |
parser.add_argument('--verbose', action='store_true', help='report mAP by class')
|
249 |
opt = parser.parse_args()
|
250 |
-
opt.save_json = opt.save_json or opt.data.endswith(
|
251 |
opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
|
252 |
print(opt)
|
253 |
|
|
|
247 |
parser.add_argument('--augment', action='store_true', help='augmented inference')
|
248 |
parser.add_argument('--verbose', action='store_true', help='report mAP by class')
|
249 |
opt = parser.parse_args()
|
250 |
+
opt.save_json = opt.save_json or opt.data.endswith('coco.yaml')
|
251 |
opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
|
252 |
print(opt)
|
253 |
|