簡體   English   中英

未找到數據集.. 缺少路徑.. 使用 yolo v8 訓練 model

[英]Dataset not found .. Missing Path .. while training the model using yolo v8

代碼單元格-

    from ultralytics import YOLO
    model = YOLO("yolov8n.pt")
    results =  model.train(data="/workspace/awadh/nvidia/apis_mellifera/v8/datasets/Apis_mellifera_IIT_front_view-1/data.yaml", epochs=200, batch=16, imgsz=640)

錯誤顯示-

/usr/local/lib/python3.8/dist-packages/ultralytics/yolo/data/utils.py in check_dataset_yaml(dataset, autodownload)
    232                 LOGGER.warning(msg)
    233             else:
--> 234                 raise FileNotFoundError(msg)
    235             t = time.time()
    236             if s.startswith('http') and s.endswith('.zip'):  # URL

FileNotFoundError: 
Dataset '/workspace/awadh/nvidia/apis_mellifera/v8/datasets/Apis_mellifera_IIT_front_view-1/data.yaml' not found ⚠️, missing paths ['/workspace/awadh/nvidia/Species_detection/Apis_Mellifera/y8/datasets/Apis_mellifera_IIT_front_view-1/valid/images']

因此,同一行代碼在 google colab 中工作,但當我使用 jupyter 時它不起作用。數據集存在於文件位置,但此錯誤出現。

根據官方python使用源,發布8.0.20:

from ultralytics.yolo.engine.model import YOLO

model = YOLO("/path/to/your/model.pt")
results = model.predict(stream=True, source='ultralytics/ultralytics/assets', save=True) # source already setup

這發生在我使用 ultralytics 8.0.22 的雲 GPU 實例上。 降級到 ultralytics 8.0.20 為我解決了這個問題,運行 pytorch/pytorch Docker 圖像。

pip uninstall ultralytics
pip install ultralytics==8.0.20

但是,這個問題在我運行 MacOS Ventura (Apple Silicon) 的本地機器上仍然存在。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM