You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
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
oroxford 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
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.
The text was updated successfully, but these errors were encountered: