Hi there,
I need to re-train an SSD model pre-trained on COCO in order to make it
recognize persons (which is already one of the COCO classes) and another custom
object (a specific mug).
I have the labelled images and the .rec files for the "mug" object. I was
wondering if it's possible to fine-tune the model only on the new class, and to
keep the original network weights for "person".
In the .lst files relative to the new "mug" class, the object is identified by
the ID 0, so I was thinking maybe I should do something like:
net = gcv.model_zoo.get_model("ssd_512_mobilenet1.0_coco", pretrained=True)
classes = ["mug", "person"]
net.reset_class(classes=['mug', person'], reuse_weights={'person':
'person'})
would that be correct?
---
[Visit
Topic](https://discuss.mxnet.apache.org/t/finetune-ssd-model-while-keeping-model-weights-for-one-class/6662/1)
or reply to this email to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.mxnet.apache.org/email/unsubscribe/0ce1e22461ea571b73912e41eec7a0b3419764f7e075563e76c24108349c1d70).
