Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

YOLOv5 CPU Export Benchmarks #6613

Merged
merged 40 commits into from
Feb 12, 2022
Merged
Changes from 1 commit
Commits
Show all changes
40 commits
Select commit Hold shift + click to select a range
8c3f233
Add benchmarks.py
glenn-jocher Jan 25, 2022
9a52627
Update
glenn-jocher Jan 25, 2022
d9933ad
Add requirements
glenn-jocher Jan 25, 2022
e03ebe0
Merge master
glenn-jocher Jan 25, 2022
818197c
Updates
glenn-jocher Jan 25, 2022
93bd69e
Updates
glenn-jocher Jan 25, 2022
83b7e07
Updates
glenn-jocher Jan 26, 2022
aeb6ee4
Updates
glenn-jocher Jan 26, 2022
89dbad3
Updates
glenn-jocher Jan 26, 2022
7dfd503
Updates
glenn-jocher Jan 26, 2022
1dd98d6
Merge master
glenn-jocher Jan 26, 2022
7bb0612
dataset autodownload from root
glenn-jocher Jan 26, 2022
b84c8d4
Merge master
glenn-jocher Feb 1, 2022
e69df23
Merge master
glenn-jocher Feb 1, 2022
a691077
Merge master
glenn-jocher Feb 1, 2022
21cf729
Merge master
glenn-jocher Feb 1, 2022
b17c125
Merge master
glenn-jocher Feb 1, 2022
d3360c5
Merge master
glenn-jocher Feb 3, 2022
0cfd8ea
Update
glenn-jocher Feb 3, 2022
7cf5f65
Redirect to /dev/null
glenn-jocher Feb 3, 2022
c3c575e
sudo --help
glenn-jocher Feb 3, 2022
7318376
Cleanup
glenn-jocher Feb 3, 2022
bab9cba
Merge master
glenn-jocher Feb 3, 2022
028d6ab
Merge master
glenn-jocher Feb 4, 2022
b82e369
Add `DATASETS_DIR` global in general.py
glenn-jocher Feb 9, 2022
61ee2d6
Merge master
glenn-jocher Feb 10, 2022
a7fbd4a
Add exports pd df
glenn-jocher Feb 11, 2022
f3231ca
Merge master
glenn-jocher Feb 11, 2022
8097ecc
Updates
glenn-jocher Feb 11, 2022
d5f8eb3
Updates
glenn-jocher Feb 11, 2022
257186c
Updates
glenn-jocher Feb 11, 2022
1f3e9f1
Cleanup
glenn-jocher Feb 11, 2022
8310a17
dir handling fix
glenn-jocher Feb 11, 2022
b3f423a
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Feb 11, 2022
ec6659f
Cleanup
glenn-jocher Feb 11, 2022
1c4b7c7
Merge remote-tracking branch 'origin/updates/benchmarks' into updates…
glenn-jocher Feb 11, 2022
4733766
Cleanup2
glenn-jocher Feb 11, 2022
4efb565
Cleanup3
glenn-jocher Feb 11, 2022
6adb9cd
Merge master
glenn-jocher Feb 12, 2022
6edbf9f
Cleanup model_type
glenn-jocher Feb 12, 2022
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Merge master
  • Loading branch information
glenn-jocher committed Feb 1, 2022
commit 21cf7291906cf5b7f199c9185ecedb816b78b829
2 changes: 1 addition & 1 deletion utils/benchmarks.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def run(weights=ROOT / 'yolov5s.pt', # weights path
y, t = [], time.time()
formats = 'torch', 'torchscript', 'onnx', 'openvino', 'engine', 'coreml', \
'saved_model', 'pb', 'tflite', 'edgetpu', 'tfjs'
for f in formats[:4]:
for f in formats[:]:
try:
w = weights if f == 'torch' else export.run(weights=weights, imgsz=[imgsz], include=[f], device='cpu')[-1]
result = val.run(data=data, weights=w, imgsz=imgsz, batch_size=batch_size, plots=False, device='cpu')
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.