Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-24 14:11:31 -08:00
parent 2ee0d0c714
commit cdc382e313
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -21,7 +21,7 @@ def test(cfg,
# Initialize/load model and set device # Initialize/load model and set device
if model is None: if model is None:
device = torch_utils.select_device(opt.device, batch_size=batch_size) device = torch_utils.select_device(opt.device, batch_size=batch_size)
verbose = True verbose = opt.task == 'test'
# Remove previous # Remove previous
for f in glob.glob('test_batch*.jpg'): for f in glob.glob('test_batch*.jpg'):