提问人:RD Smart AI Solution 提问时间:8/22/2023 最后编辑:Tim RRD Smart AI Solution 更新时间:8/23/2023 访问量:186
在 Colab 中运行单元以在高 epoch 中训练模型 YOLOv8 时出错
Error running cell in Colab to train model YOLOv8 in high epochs
问:
我遇到了在 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 Found
FileNotFoundError : Image Not Found
我已经检查了图像路径并验证它们是可以正常打开的正常图像。
我仍然不知道导致错误的确切原因以及如何修复它。
答:
代码正在查找指定位置中的文件:
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 文件,我不确定。
不过,这是您当前的问题,在提供该文件之前,您的代码将继续失败。
评论