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

updates

This commit is contained in:
Glenn Jocher 2019-04-05 15:54:59 +02:00
parent 325b1ba4bc
commit 5e79810e69
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -137,7 +137,7 @@ def test(
mp, mr, map, mf1 = p.mean(), r.mean(), ap.mean(), f1.mean() mp, mr, map, mf1 = p.mean(), r.mean(), ap.mean(), f1.mean()
# Print results # Print results
pf = ('%20s' + '%10.3g' * 6) # print format pf = '%20s' + '%10.3g' * 6 # print format
print(pf % ('all', seen, nt.sum(), mp, mr, map, mf1), end='\n\n') print(pf % ('all', seen, nt.sum(), mp, mr, map, mf1), end='\n\n')
# Print results per class # Print results per class

2
train.py
View File

@ -33,7 +33,7 @@ def train(
img_size = 608 # initiate with maximum multi_scale size img_size = 608 # initiate with maximum multi_scale size
num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174
else: else:
torch.backends.cudnn.benchmark = True # unsuitable for multiscale pass # torch.backends.cudnn.benchmark = True # unsuitable for multiscale
# Configure run # Configure run
train_path = parse_data_cfg(data_cfg)['train'] train_path = parse_data_cfg(data_cfg)['train']