Skip to content

Commit

Permalink
Add ONNXRuntime support statement (#174)
Browse files Browse the repository at this point in the history
* Fixing lint

* Update onnxruntime C++ interface support

* Fixing typo

* Rename to Thanks

* Minor fix
  • Loading branch information
zhiqwang authored Sep 23, 2021
1 parent 7ab6c96 commit ed4c4eb
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 11 deletions.
15 changes: 8 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,14 @@

## 🆕 What's New

- Add `TVM` compile and inference notebooks. *Feb. 5, 2021*.
- Add graph visualization tools. *Nov. 21, 2020*.
- Support exporting to `ONNX`, and inferring with `ONNXRuntime` interface. *Nov. 17, 2020*.
- Refactor YOLO modules and support *dynamic shape/batch* inference. *Nov. 16, 2020*.
- Add `TorchScript` C++ inference example. *Nov. 4, 2020*.
- Support inferring with `LibTorch` C++ interface. *Oct. 10, 2020*.
- Support exporting to `TorchScript` model. *Oct. 8, 2020*.
- *Sep. 24, 2021*. Add `ONNXRuntime` C++ interface example. Thanks to [itsnine](https://github.com/itsnine).
- *Feb. 5, 2021*. Add `TVM` compile and inference notebooks.
- *Nov. 21, 2020*. Add graph visualization tools.
- *Nov. 17, 2020*. Support exporting to `ONNX`, and inferencing with `ONNXRuntime` Python interface.
- *Nov. 16, 2020*. Refactor YOLO modules and support *dynamic shape/batch* inference.
- *Nov. 4, 2020*. Add `TorchScript` C++ inference example.
- *Oct. 10, 2020*. Support inferencing with `LibTorch` C++ interface.
- *Oct. 8, 2020*. Support exporting to `TorchScript` model.

## 🛠️ Usage

Expand Down
8 changes: 4 additions & 4 deletions deployment/onnxruntime/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ The ONNXRuntime inference for `yolort`, both GPU and CPU are supported.

```bash
./yolort_onnx [--image ../../../test/assets/zidane.jpg]
[--model_path ../../../test/tracing/yolov5s.onnx]
[--class_names ../../../notebooks/assets/coco.names]
[--gpu] # GPU switch, which is optional, and set False as default
```
[--model_path ../../../notebooks/yolov5s.onnx]
[--class_names ../../../notebooks/assets/coco.names]
[--gpu] # GPU switch, which is optional, and set False as default
```

0 comments on commit ed4c4eb

Please sign in to comment.