https://github.com/jahongir7174/yolov8-onnx
YOLOv8 inference using ONNX Runtime
https://github.com/jahongir7174/yolov8-onnx
object-detection onnx onnxruntime onnxruntime-gpu opencv-python python yolov8
Last synced: 3 months ago
JSON representation
YOLOv8 inference using ONNX Runtime
- Host: GitHub
- URL: https://github.com/jahongir7174/yolov8-onnx
- Owner: jahongir7174
- Created: 2023-10-10T02:35:12.000Z (over 1 year ago)
- Default Branch: master
- Last Pushed: 2023-10-10T02:37:52.000Z (over 1 year ago)
- Last Synced: 2025-03-24T12:46:58.750Z (3 months ago)
- Topics: object-detection, onnx, onnxruntime, onnxruntime-gpu, opencv-python, python, yolov8
- Language: Python
- Homepage:
- Size: 15.8 MB
- Stars: 7
- Watchers: 0
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
YOLOv8 inference using ONNX Runtime
### Installation
```
conda create -n ONNX python=3.8.10
conda activate ONNX
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia
pip install opencv-python
pip install onnx
pip install onnxsim
pip install onnxruntime-gpu
```
### Train
* Follow [YOLOv8](https://github.com/jahongir7174/YOLOv8-pt) for training
### Export to ONNX
* Configure your model path in `main.py` for exporting
* Run `python main.py --export`
### Test
* Configure your exported onnx model path and image path in `main.py` for testing
* Run `python main.py --test` for testing
#### Reference
* https://github.com/jahongir7174/YOLOv8-pt