赞
踩
每次代码更新版本都要费眼睛去适应,写个教程方便他人
这里是写给有YOLOv5基础的,因为v5的新手教程已经很多了,v8和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是一模一样的,不用改
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
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。