如何在colab配置YOLO v8并训练自己的数据集

最新推荐文章于 2024-03-29 23:17:22 发布

偶就是偶

最新推荐文章于 2024-03-29 23:17:22 发布

阅读量1.4k

点赞数

文章标签： YOLO 深度学习人工智能 Powered by 金山文档

本文链接：https://blog.csdn.net/wcc960828/article/details/129454102

版权

训练前的参数配置

在 /content/ultralytics/ultralytics/yolo/cfg/default.yaml中修改了以下参数：但似乎不改也没事

# Train settings -------------------------------------------------------------------------------------------------------
model:/content/ultralytics/yolov8s.pt  # path to model file, i.e. yolov8n.pt, yolov8n.yaml
data: /content/Letters.v1i.yolov5pytorch.zip/data.yaml # path to data file, i.e. coco128.yaml
epochs: 100  # number of epochs to train for
patience: 50  # epochs to wait for no observable improvement for early stopping of training
batch: 16  # number of images per batch (-1 for AutoBatch)
imgsz: 640  # size of input images as integer or w,h
save: True  # save train checkpoints and predict results
save_period: -1 # Save checkpoint every x epochs (disabled if < 1)