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

Add segmentation + classification head fine-tuning example #912

Open
qubvel opened this issue Aug 23, 2024 · 0 comments
Open

Add segmentation + classification head fine-tuning example #912

qubvel opened this issue Aug 23, 2024 · 0 comments
Labels

Comments

@qubvel
Copy link
Collaborator

qubvel commented Aug 23, 2024

It would be great to have more examples for fine-tuning in the library!

Similar to current examples for binary-segmentaion and multi-label segmentation would be great to have example for segmentation + classification head. The same camvid or oxford pet dataset can be used. As a classification target for these datasets, the "existence" of the class on the image can be used.

The example should showcase how to

  • fine-tune a model with pytorch-lightning (or any other fine-tuning framework, even a plain pytorch)
  • use loss function for segmentation and classification
  • compute metrics for classification and segmentation heads
  • visualize results

Docs on how to create a model with classification head.

In case anyone wants to work on this you are welcome! Just notify in this issue, and then ping me to review a PR when it's ready.

@qubvel qubvel added good first issue Good for newcomers examples labels Aug 23, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant