zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update benchmarks significant digits (#7103)

This commit is contained in:
Glenn Jocher 2022-03-22 18:02:35 +01:00 committed by GitHub
parent 05aae17333
commit ee0b3b2a95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/benchmarks.py
View File

@ -67,7 +67,7 @@ def run(weights=ROOT / 'yolov5s.pt', # weights path
result = val.run(data, w, batch_size, imgsz, plots=False, device=device, task='benchmark', half=half) result = val.run(data, w, batch_size, imgsz, plots=False, device=device, task='benchmark', half=half)
metrics = result[0] # metrics (mp, mr, map50, map, *losses(box, obj, cls)) metrics = result[0] # metrics (mp, mr, map50, map, *losses(box, obj, cls))
speeds = result[2] # times (preprocess, inference, postprocess) speeds = result[2] # times (preprocess, inference, postprocess)
y.append([name, metrics[3], speeds[1]]) # mAP, t_inference y.append([name, round(metrics[3], 4), round(speeds[1], 2)]) # mAP, t_inference
except Exception as e: except Exception as e:
LOGGER.warning(f'WARNING: Benchmark failure for {name}: {e}') LOGGER.warning(f'WARNING: Benchmark failure for {name}: {e}')
y.append([name, None, None]) # mAP, t_inference y.append([name, None, None]) # mAP, t_inference