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
Hydranet - separate prediction heads per class #11683
Comments
Or more precisely, how could I go about adding a classification head, to the SwinTransformer Instance segmentation model? |
Seeing some promising bits for multiple models in extract_model.py of MMDeploy |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Describe the feature
Motivation
I'd love to be able to train an instance segmentation and semantic segmentation model with the same backbone.
Or train a model with the same backbone, but a prediction head per class, so that I could update their weights independently. I guess I could already achieve this by doing something like freezing the backbone, exporting to ONNX, splitting the model appropriately, the same backbone repeatedly, but different heads as needed.
Do any models in MMDetection already make it easy to get started with this?
Any guidance at all would be very helpful.
The text was updated successfully, but these errors were encountered: