从YOLOv5到YOLOv8(0.52)

2023-11-01

每次代码更新版本都要费眼睛去适应,写个教程方便他人


前言

这里是写给有YOLOv5基础的,因为v5的新手教程已经很多了,v8和v5的源码形式上差不多,只是调用的时候有点区别而已。


YOLOv8 github源码

一、直接用v5的环境

v8的环境和v5没什么主要区别,待会儿跑一遍,缺什么补什么就好。

二、代码文件

下载源文件,不用全部解压,只需要将子文件ultralytics到你的项目文件夹下。例如:
文件示例
这里的main_train.py文件是我自己创的,为的就是运行训练,而且代码也相当简单:

from ultralytics import YOLO

if __name__ == '__main__':
    # 加载模型
    # model = YOLO("yolov8n.yaml")
    model = YOLO("./yolov8n.pt")  # 用初始化权重
    results = model.train(data="./coco128.yaml")  # 载入数据集
    # results = model.val()  # 在验证集上评估模型性能

这里的data.yaml跟YOLOv5是一模一样的,不用改

二、超参数修改

1.yaml文件

v8把所有的参数放在ultralytics\yolo\cfg\default.yaml文件下了,相当于把v5的cfg.yaml和paser堆到了一起,同时还包含着detect、export和tracing(这个v5没有)的参数。总之就是弄到一块去了:

# Ultralytics YOLO 									
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

从YOLOv5到YOLOv8(0.52) 的相关文章

随机推荐

Powered by Hwhale