Can't we do better than destroy the entire classification head when the annotations don't match?

weecology / DeepForest

Python Package for Airborne RGB machine learning

https://deepforest.readthedocs.io/

MIT License

521 stars 176 forks source link

Can't we do better than destroy the entire classification head when the annotations don't match? #798

Open bw4sz opened 1 month ago

bw4sz commented 1 month ago

I think we can provide users with more insight than just a random instantiation of the backbone when we are running multi-class and retraining to add new labels.

https://deepforest.readthedocs.io/en/latest/advanced_features/multi_species.html

We could grab the classification head from the release model. We could get just the top layers. I think this was one of the workarounds we identified and then didn't return to.