roboflow / notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.
https://roboflow.com/models
5.09k stars 788 forks source link

Get error on classification training when using custom dataset in yolo v8 #92

Closed Sayed049 closed 1 year ago

Sayed049 commented 1 year ago

in detection and segmentation 'custom training' i use (data='data.yaml'), see in the image below Screenshot_1

but in classification 'custom training' i use (data='data.yaml') bt get the error, see in the image below
Screenshot_2 i your notebook u tell data=dataset.location bt i dont understand where did i get dataset.location ?

github-actions[bot] commented 1 year ago

👋 Hello @Sayed049, thank you for leaving an issue on Roboflow Notebooks.

🐞 Bug reports

If you are filing a bug report, please be as detailed as possible. This will help us more easily diagnose and resolve the problem you are facing. To learn more about contributing, check out our Contributing Guidelines.

If you require support with custom code that is not part of Roboflow Notebooks, please reach out on the Roboflow Forum or on the GitHub Discussions page associated with this repository.

💬 Get in touch

Do you have more questions about Roboflow that we haven't responded to yet? Feel free to ask them on the Roboflow Discuss forum. Our developer advocates and community team actively respond to questions there.

To ask questions about Notebooks, head over to the GitHub Discussions section of this repository.

SkalskiP commented 1 year ago

Here is the command we use for training in our notebook: !yolo task=classify mode=train model=yolov8n-cls.pt data={dataset.location} epochs=50 imgsz=128. For classification, you do not pass data.yaml. You pass a directory containing images.

I am closing the issue. Feel free to reopen in case you'll have more questions.

Sayed049 commented 1 year ago

i pass the directory that contain images but it still show errors Screenshot_1 the directory is in the drive, here is the screen shot Screenshot_2