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

Support upstream yolov5 v6 release #194

Merged
merged 13 commits into from
Oct 10, 2021
Merged

Support upstream yolov5 v6 release #194

merged 13 commits into from
Oct 10, 2021

Conversation

zhiqwang
Copy link
Owner

@zhiqwang zhiqwang commented Oct 8, 2021

Following the upstream YOLOv5 release v6.0 ultralytics/yolov5#5141 . For release v6.0, use the snippet below

from yolort.models import YOLOv5

# 'yolov5s.pt' is downloaded from https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5s.pt
ckpt_path_from_ultralytics = "yolov5s.pt"
model = YOLOv5.load_from_yolov5(ckpt_path_from_ultralytics, version="r6.0")

model.eval()
img_path = "test/assets/bus.jpg"
predictions = model.predict(img_path)

Whereas for release v4.0 and v5.0, just change the version to version="r4.0" like below

# 'yolov5s.pt' is downloaded from https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
# Or from https://github.com/ultralytics/yolov5/releases/download/v4.0/yolov5s.pt
ckpt_path_from_ultralytics = "yolov5s.pt"
model = YOLOv5.load_from_yolov5(ckpt_path_from_ultralytics, version="r4.0")

model.eval()
img_path = "test/assets/bus.jpg"
predictions = model.predict(img_path)

@zhiqwang zhiqwang added the enhancement New feature or request label Oct 8, 2021
@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

✅ zhiqwang
❌ pre-commit-ci[bot]
You have signed the CLA already but the status is still pending? Let us recheck it.

@codecov
Copy link

codecov bot commented Oct 8, 2021

Codecov Report

Merging #194 (8af0cb5) into master (d1f02f6) will not change coverage.
The diff coverage is n/a.

❗ Current head 8af0cb5 differs from pull request most recent head a3953e5. Consider uploading reports for the commit a3953e5 to get more accurate results
Impacted file tree graph

@@           Coverage Diff           @@
##           master     #194   +/-   ##
=======================================
  Coverage   96.49%   96.49%           
=======================================
  Files          10       10           
  Lines         599      599           
=======================================
  Hits          578      578           
  Misses         21       21           
Flag Coverage Δ
unittests 96.49% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.


Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update d1f02f6...a3953e5. Read the comment docs.

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@zhiqwang zhiqwang merged commit 1c616c7 into master Oct 10, 2021
@zhiqwang zhiqwang deleted the update-yolov5-v6 branch October 10, 2021 05:43
This was referenced Oct 10, 2021
@zhiqwang zhiqwang added the API Library use interface label Oct 14, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
API Library use interface enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants