在 Colab 中运行单元以在高 epoch 中训练模型 YOLOv8 时出错

Error running cell in Colab to train model YOLOv8 in high epochs

提问人:RD Smart AI Solution 提问时间:8/22/2023 最后编辑:Tim RRD Smart AI Solution 更新时间:8/23/2023 访问量:186

问:

我遇到了在 Colab 中运行单元以训练模型 YOLOv8 的问题。该代码将使用此代码。

import os from ultralytics
import YOLO
model = YOLO("yolov8x.yaml")
results = model.train(data=os.path.join(ROOT_DIR, "google_colab_config.yaml"), epochs=60)

这列火车上总共有 15863 张图像。当以 1、5 或 21 个 epoch 运行时没有问题,模型可以正常运行,但是当我开始使用更多数量的 epoch 时,例如,我使用 70 个 epoch。然后,在第 38 个时期会出现一个错误,说后面跟着该图像的路径,或者我使用 60 个时期,它会在第 35 个时期出错,说后面跟着该图像的路径FileNotFoundError : Image Not FoundFileNotFoundError : Image Not Found

我已经检查了图像路径并验证它们是可以正常打开的正常图像。

我仍然不知道导致错误的确切原因以及如何修复它。

错误消息的屏幕截图

python google-colaboratory training-data yolov8 filenotfounderror

评论


答:

0赞 J Wylie 8/22/2023 #1

代码正在查找指定位置中的文件:

rootdir/google_colab_config.yaml

您是否将该 google_colab_config.yaml 文件上传到 colab?

旁边有一个文件资源管理器,可以创建或上传文件。

您键入的代码可能需要更改,或者您需要将“ROOT_DIR”变量链接到 colab 存储文件的位置(通常为 /content 文件夹)

获取 google_colab_config.yaml 文件,将其上传到您的 colab 工作簿,然后右键单击它并选择“复制路径”,然后将其粘贴到当前代码的位置,因此它来自:

os.path.join(ROOT_DIR, "google_colab_config.yaml"

自:

import os 
from ultralytics import YOLO 
model = YOLO("yolov8x.yaml") 
results = model.train(data="/content/google_colab_config.yaml", epochs=60)

不过,在您上传该文件之前,它仍然无法正常工作,如果这是某种课程,则需要联系教师或您的同学以获取该文件。

您作业的另一部分可能是创建那个 google colab config yaml 文件,我不确定。

不过,这是您当前的问题,在提供该文件之前,您的代码将继续失败。