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 yolov5x and yolov5x6 models #261

Merged
merged 5 commits into from
Dec 29, 2021

Conversation

ShiquanYu
Copy link
Contributor

@ShiquanYu ShiquanYu commented Dec 29, 2021

Support yolov5x and yolov5x6 models

from yolort.models import YOLOv5

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

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

@codecov
Copy link

codecov bot commented Dec 29, 2021

Codecov Report

Merging #261 (0934555) into main (5dc8996) will not change coverage.
The diff coverage is n/a.

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

@@           Coverage Diff           @@
##             main     #261   +/-   ##
=======================================
  Coverage   93.62%   93.62%           
=======================================
  Files          11       11           
  Lines         675      675           
=======================================
  Hits          632      632           
  Misses         43       43           
Flag Coverage Δ
unittests 93.62% <ø> (ø)

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 5dc8996...b435234. Read the comment docs.

@zhiqwang zhiqwang changed the title [WIP]yolov5-x support Support yolov5x and yolov5x6 models Dec 29, 2021
@zhiqwang zhiqwang added the enhancement New feature or request label Dec 29, 2021
@zhiqwang zhiqwang merged commit ac6ec43 into zhiqwang:main Dec 29, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants