YOLOV8小目标检测
前言::
yolo版出现很多,基本大同小异
但是这些差异让我们考虑在实验中使用哪个版本会比较好!
在对小目标检测的过程中,yolov7相比yolov8性能更加好。
如果我们还是想使用yolov8,也是可以实现对小目标进行检测
YOLOV8更换yaml文件||实现小目标检测功能
python
from ultralytics import YOLO
if __name__ == '__main__':
# 加载预训练模型,在该模型基础上,训练目标检测的模型
model = YOLO('yolov8s-p2.yaml').load('yolov8s.pt')
#训练自定义数据集,数据配置保存在data.yaml中
model.train(data='data.yaml',epochs=100,batch=64,workers=0,device=0)
#使用验证集验证效果
model.val()