Using a pretrained model as backbone #268
Replies: 3 comments
-
My idea is that I want to detect objects similar to the data I used the model to classify, so I thought of reusing the weights of my classification models for the detection model,rather than training an OD model from scratch. |
Beta Was this translation helpful? Give feedback.
-
@mrinath123 there's no builtin support to load a classifier model into the backbone of the effdet model from a checkpoint -- it only supports loading pretrained weights from the timm pretrained weights if the pretrained_backbone flag is set... However, it'd be fairly straightforward after the effdet model is created, to load state dict from your pretrained model |
Beta Was this translation helpful? Give feedback.
-
@rwightman THanks, but es I was not specifically telling about classification models, we could use any type of backbone trained on any type of task, classification was just an example tried by me. |
Beta Was this translation helpful? Give feedback.
-
HI!
I created a binary classification model using "tf_efficientnet_b4_ns", the model class is below
I trained this model, giving some good accuracy/AUC, Now I am trying to use this model with its learned weights as the backbone of an effdet model, how to do that?
Beta Was this translation helpful? Give feedback.
All reactions