You are viewing a plain text version of this content. The canonical link for it is here.

Posted to discuss-archive@mxnet.apache.org by Lews Therin511 via MXNet Forum <mx...@discoursemail.com.INVALID> on 2020/10/16 10:54:30 UTC

[MXNet Forum] Finetune SSD model while keeping model weights for one class


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).