You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
**What it is.** Yet another implementation of Ultralytics's [YOLOv5](https://github.com/ultralytics/yolov5). `yolort` aims to make the training and inference of the object detection integrate more seamlessly together. `yolort` now adopts the same model structure as the official YOLOv5. The significant difference is that we adopt the dynamic shape mechanism, and within this, we can embed both pre-processing (`letterbox`) and post-processing (`nms`) into the model graph, which simplifies the deployment strategy. In this sense, `yolort` makes it possible to be deployed more friendly on `LibTorch`, `ONNXRuntime`, `TVM` and so on.
-*Nov. 4, 2020*. Add `LibTorch` C++ inference example.
55
55
-*Oct. 8, 2020*. Support exporting to `TorchScript` model.
56
56
57
-
## :hammer_and_wrench: Usage
57
+
## 🛠️ Usage
58
58
59
59
There are no extra compiled components in `yolort` and package dependencies are minimal, so the code is very simple to use.
60
60
@@ -135,17 +135,30 @@ We provide a [notebook](notebooks/inference-pytorch-export-libtorch.ipynb) to de
135
135
136
136
On the `ONNXRuntime` front you can use the [C++ example](deployment/onnxruntime), and we also provide a tutorial [export-onnx-inference-onnxruntime](notebooks/export-onnx-inference-onnxruntime.ipynb) for using the `ONNXRuntime`.
137
137
138
-
## :art: Model Graph Visualization
138
+
## 🎨 Model Graph Visualization
139
139
140
140
Now, `yolort` can draw the model graph directly, checkout our [model-graph-visualization](notebooks/model-graph-visualization.ipynb) notebook to see how to use and visualize the model graph.
141
141
142
142
<ahref="notebooks/assets/yolov5_graph_visualize.svg"><imgsrc="notebooks/assets/yolov5_graph_visualize.svg"alt="YOLO model visualize"width="500"/></a>
143
143
144
-
## :mortar_board: Acknowledgement
144
+
## 🎓 Acknowledgement
145
145
146
146
- The implementation of `yolov5` borrow the code from [ultralytics](https://github.com/ultralytics/yolov5).
147
147
- This repo borrows the architecture design and part of the code from [torchvision](https://github.com/pytorch/vision).
148
148
149
-
## :+1: Contributing
149
+
## 📖 Citing yolort
150
+
151
+
If you use yolort in your publication, please cite it by using the following BibTeX entry.
152
+
153
+
```bibtex
154
+
@Misc{yolort2021,
155
+
author = {Zhiqiang Wang, Fidan Kharrasov},
156
+
title = {yolort: A runtime stack for object detection on specialized accelerators},
See the [CONTRIBUTING](.github/CONTRIBUTING.md) file for how to help out. BTW, leave a :star2: if you liked it, and this is the easiest way to support us :)
0 commit comments