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

Constructs YOLOv5 model with TorchVision MobileNetV3 backbone #342

Merged
merged 10 commits into from
Mar 5, 2022

Conversation

zhiqwang
Copy link
Owner

@zhiqwang zhiqwang commented Mar 5, 2022

We use TorchVision layout to construct a high resolution YOLOv5 model with TorchVision pre-trained MobileNetV3-Small FPN backbone.

import torch
from yolort.models.yolo_lite import yolov5_mobilenet_v3_small_fpn

model = yolov5_mobilenet_v3_small_fpn()
model = model.eval()

images = torch.rand(4, 3, 640, 640)
out = model(images)

@zhiqwang zhiqwang added the enhancement New feature or request label Mar 5, 2022
@codecov
Copy link

codecov bot commented Mar 5, 2022

Codecov Report

Merging #342 (d67cc33) into main (de11966) will not change coverage.
The diff coverage is n/a.

Impacted file tree graph

@@           Coverage Diff           @@
##             main     #342   +/-   ##
=======================================
  Coverage   95.46%   95.46%           
=======================================
  Files          11       11           
  Lines         771      771           
=======================================
  Hits          736      736           
  Misses         35       35           
Flag Coverage Δ
unittests 95.46% <ø> (ø)

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 de11966...d67cc33. Read the comment docs.

@zhiqwang zhiqwang added the API Library use interface label Mar 5, 2022
@zhiqwang zhiqwang merged commit b96c225 into main Mar 5, 2022
@zhiqwang zhiqwang deleted the mobilenetv3 branch March 5, 2022 07:43
zhiqwang added a commit that referenced this pull request Mar 8, 2022
* Add yolov5n and yolov5lite for training with ultralytics

* Apply pre-commit

* Minor fixes for docstrings

* Use frcnn layout

* Fix configurations for mobilenetv3 with yolov5

* Apply pre-commit

* Move yolov5lite into yolort.models

* Minor fixes

* Apply pre-commit

* Change to mobilenet_v3_small
zhiqwang added a commit that referenced this pull request Mar 8, 2022
* Add yolov5n and yolov5lite for training with ultralytics

* Apply pre-commit

* Minor fixes for docstrings

* Use frcnn layout

* Fix configurations for mobilenetv3 with yolov5

* Apply pre-commit

* Move yolov5lite into yolort.models

* Minor fixes

* Apply pre-commit

* Change to mobilenet_v3_small
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.

1 participant