当前位置:   article > 正文

从YOLOv5到YOLOv8(0.52)_yolov8和yolov5环境

yolov8和yolov5环境

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


前言

这里是写给有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()  # 在验证集上评估模型性能
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

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

二、超参数修改

1.yaml文件

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

# Ultralytics YOLO 
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/150802
推荐阅读
相关标签