Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Finetune YOLOv8 to detect shapes #145

Open
JanelleCai opened this issue Jan 7, 2025 · 0 comments
Open

Finetune YOLOv8 to detect shapes #145

JanelleCai opened this issue Jan 7, 2025 · 0 comments

Comments

@JanelleCai
Copy link
Contributor

  • using the shapes dataset (Create a shapes dataset #143), finetune a YOLOv8 model to detect the shapes
  • if you are stuck on where to start, you can take a look at this file, which is what we used to finetune YOLOv8 to detect buoys. keep in mind, this notebook loads in the buoys dataset from roboflow - the main change you'll need to make is to instead load in the shapes dataset.
  • you can decide if the model should detect both color + shape versus just the shape.
    • if the data includes all colors and shapes specified in Create a shapes dataset #143 , you can train the model to detect both color + shape (and this is probably the better option).
    • however, if the data is missing certain colors, you should train it to detect shape only (and then write another node that detects the color manually using HSV values).
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant