首先,这篇文章是针对不修改源码,只是单纯希望运行YOLOv8对自己数据集进行测试,完成论文部分对比实验部分结果的朋友们,如果需要修改源码仍是需要按照一步步安装源码安装环境进行操作的。
由于本人实验时是租借云服务器的,捣鼓好久仍然没有完全安装下源码和环境。在查阅资料时无意发现其实如果只是希望借网络跑个数据集无需把整个代码都下载的
官方网址:
https://colab.research.google.com/github/ultralytics/ultralytics/blob/main/examples/tutorial.ipynb
首先,安装一下yolov8特有的集成为第三方包的包:
pip install ultralytics
之后进入python部分检查机器
import ultralytics
ultralytics.checks()
最后只需要在命令行再输入
yolo train model=yolov8xx .pt data=xxx .yaml epochs=xxx imgsz=xxx device=x workers=x
其中x的部分都是按照自己喜好进行修改的部分
我自己的话是
yolo train model=yolov8n.pt data=fire.yaml epochs=100 imgsz=640 device=0 workers=8
最后等待结果即可
不得不说,集成为第三方包真是非常方便的操作,训练好的预测直接yolo detect就可以完成,真的是贴心的YOLO
YOLOv8不修改源码训练自己数据集最简单做法
于 2023-04-07 20:11:03 首次发布