Comments (18)
We have updated the yolov5 tensorrt according to the v2.0 release of this repo.
And made speed test on my machine.
Models | Device | BatchSize | Mode | Input Shape(HxW) | FPS |
---|---|---|---|---|---|
YOLOv5-s | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 142 |
YOLOv5-s | Xeon E5-2620/GTX1080 | 4 | FP16 | 608x608 | 173 |
YOLOv5-s | Xeon E5-2620/GTX1080 | 8 | FP16 | 608x608 | 190 |
YOLOv5-m | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 71 |
YOLOv5-l | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 40 |
YOLOv5-x | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 27 |
please find https://github.com/wang-xinyu/tensorrtx.
@glenn-jocher could you also add a link to https://github.com/wang-xinyu/tensorrtx in your Tutorials section?
from yolov5.
I just implemented yolov5-s in my repo https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5
, and test on my machine. yolov5-m, yolov5-l, etc, will come out soon.
Models | Device | BatchSize | Mode | Input Shape(HxW) | FPS |
---|---|---|---|---|---|
YOLOv3-spp(darknet53) | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 38.5 |
YOLOv4(CSPDarknet53) | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 35.7 |
YOLOv5-s | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 167 |
YOLOv5-s | Xeon E5-2620/GTX1080 | 4 | FP16 | 608x608 | 182 |
YOLOv5-s | Xeon E5-2620/GTX1080 | 8 | FP16 | 608x608 | 186 |
from yolov5.
@wang-xinyu did a great TensorRT implementation of our https://github.com/ultralytics/yolov3 repo here (which supports both YOLOv3 and YOLOv4), he might best answer this question.
https://github.com/wang-xinyu/tensorrtx/tree/master/yolov3-spp
from yolov5.
Update! My tensorrt implementation already updated according to this commit 364fcfd
The PANet updated.
Please find my repo https://github.com/wang-xinyu/tensorrtx
from yolov5.
@thancaocuong there is no such file.
from yolov5.
I have a python implementation here, with NMS, https://github.com/TrojanXu/yolov5-tensorrt
from yolov5.
Hello @wu-ruijie, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook , Docker Image, and Google Cloud Quickstart Guide for example environments.
If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.
If this is a custom model or data training question, please note that Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:
- Cloud-based AI systems operating on hundreds of HD video streams in realtime.
- Edge AI integrated into custom iOS and Android apps for realtime 30 FPS video inference.
- Custom data training, hyperparameter evolution, and model exportation to any destination.
For more information please visit https://www.ultralytics.com.
from yolov5.
Tensor core support would be amazing!
from yolov5.
Hi! I tested yolo5-s on cpu by directly running detect.py and the inference speed is only 3 fps.Could you please give me some advice?I want to make it 30 fps at least.
from yolov5.
@sljlp you might want to see 'Running yolov5 on CPU' #37
The default --img-size for detect.py is 640, which you can reduce significantly to get the FPS you are looking for.
from yolov5.
@sljlp one caveat is --img-size must be a multiple of the largest stride, 32. So acceptable sizes are 320, 288, 256, etc.
from yolov5.
Update: I've pushed more robust error-checking on --img-size now in 099e6f5, so if a user accidentally requests an invalid size (which is not divisible by 32), the code will warn and automatically correct the value to the nearest valid --img-size.
from yolov5.
@glenn-jocher Can you provide yolov5.weights file. I've found that to convert yolo to tensorrt, we need the weights file to use with (https://github.com/wang-xinyu/tensorrtx/)
from yolov5.
Update! My tensorrt implementation already updated according to this commit 364fcfd
The PANet updated.
Please find my repo https://github.com/wang-xinyu/tensorrtx
Thanks for sharing! Do you have plans to implement other yolov5 versions as well?
from yolov5.
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
from yolov5.
@wang-xinyu thanks, yes this is a good idea. Can you submit a PR for the README please?
EDIT: I'll add a link to the export tutorial also.
from yolov5.
https://github.com/ttanzhiqiang/onnx_tensorrt_project
from yolov5.
We have updated the yolov5 tensorrt according to the v2.0 release of this repo.
And made speed test on my machine.
Models Device BatchSize Mode Input Shape(HxW) FPS
YOLOv5-s Xeon E5-2620/GTX1080 1 FP16 608x608 142
YOLOv5-s Xeon E5-2620/GTX1080 4 FP16 608x608 173
YOLOv5-s Xeon E5-2620/GTX1080 8 FP16 608x608 190
YOLOv5-m Xeon E5-2620/GTX1080 1 FP16 608x608 71
YOLOv5-l Xeon E5-2620/GTX1080 1 FP16 608x608 40
YOLOv5-x Xeon E5-2620/GTX1080 1 FP16 608x608 27
please find https://github.com/wang-xinyu/tensorrtx.@glenn-jocher could you also add a link to https://github.com/wang-xinyu/tensorrtx in your Tutorials section?
Thx for your work, I just wonder how do u test FPS with batchsize.
Cause our video is just one img flow, every img is in a serial line, so why could u use batchsize more than 1?
from yolov5.
Related Issues (20)
- 提升训练速度 HOT 1
- is there a max limit to --imgsz ? HOT 6
- RuntimeError: The size of tensor a (24) must match the size of tensor b (20) at non-singleton dimension 2 HOT 5
- How to show count in screen using yolov5 HOT 6
- How to change annotations indices in memory without changing the dataset locally? HOT 3
- How to add a button inside the video stream of yolov5. HOT 1
- Extract feature vector from the bounding box predicted together with the coordinates and class output vector HOT 5
- augmentation in validation HOT 1
- About detect.py HOT 9
- How to close window in yolov5 detection HOT 1
- Training YoloV5n on a custom dataset, best.pt is bigger than yolov5n official size HOT 4
- Data Augmentation HOT 1
- about eval.py HOT 1
- Need advice for training a YOLOv5-obb model HOT 2
- Code doubts about the model in the detection process HOT 2
- predicting from 2D array HOT 2
- Same yolov5s training, but one over-fitting and one training is very good. HOT 2
- Hello, I have some questions about the YOLOv5 code. Could you please help me answer them? HOT 2
- Different results from train.py and val.py HOT 1
- How to change training input image size? HOT 8
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yolov5.