本文主要是介绍YOLOv8-PySide --- 基于 ultralytics 8.1.0 发行版优化 | 代码已开源,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
YOLOv8-PySide — 基于 ultralytics 8.1.0 发行版优化
Github 项目地址:https://github.com/WangQvQ/Ultralytics-PySide6
BiliBili视频地址:https://www.bilibili.com/video
页面效果
如何使用
pip install ultralytics==8.1.0 or git clone --branch v8.1.0 --single-branch https://github.com/ultralytics/ultralytics.git
pip install pyside6 chardet
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
python main.py
项目功能
- ✅ 图片推理
- ✅ 视频推理
- ✅ 摄像头推理
- ❌ RTSP 推流
- ✅ 分类任务推理
- ✅ 检测任务推理
- ✅ 分割任务推理
- ✅ 关键点任务推理
- ❌ 追踪任务推理
- ❌ 旋转框任务推理
- ✅ Pytroch (.pt) 格式模型推理
- ✅ ONNX (.onnx) 格式模型推理
- ✅ TensorRT (.engine) 格式模型推理
- ✅ 模型选择
- ✅ 置信度/阈值调整
- ✅ 延迟调整
- ✅ 保存推理结果
注意事项
- 跟踪功能未集成。
- 旋转框检测未集成。
- 打包成功可能无法运行。
- 如果想使用自己的模型,您需要先使用
ultralytics
来训练 yolov8 模型,然后将训练好的.pt/.onnx/.engine
文件放入models/*
文件夹。 - 如果模型是改进的,请将你整个项目文件导入。
- 如果选择保存结果,结果会保存在
./run
路径中。 - UI 设计文件是
home.ui
,如果修改它,您需要使用pyside6-uic home.ui > ui/home.py
命令来重新生成.py
文件。 - 资源文件是
resources.qrc
,如果您修改了默认图标,需要使用pyside6-rcc resources.qrc > ui/resources_rc.py
命令来重新生成.py
文件。
References
- PyQt5-YOLOv5
- ultralytics
- PySide6-YOLOv8
- YOLOv8-GUI-PySide6
这篇关于YOLOv8-PySide --- 基于 ultralytics 8.1.0 发行版优化 | 代码已开源的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!