Use a pretrained model adding some labels

[miguemorales]

Hi to everyone, First of all, thanks for this amazing way to use AI for begginers. Regarding my issue, i have follow the tutorial to retrain the ssd-mobilenet (in my case with the label deer) but once you retrain with that, you lose the previous labels that the model have. Is it possible for to retrain the SSD_MOBILENET_V2 creating new labels, but also keeping the previous ones. In my case i would like to train a model in which i can distinguish deers from cows from other animals, but once i retrain the model with deers, any animal is a deer for the model, that is why i would like to just add the label deer to the existing model. Thanks in advance for any help, and sorry if someone have already ask the same question, i haven't found it in the thread

[dusty-nv]

Hi @miguemorales, I think that adding a class requires re-training across all of the data. For reference, see this similar discussion about adding a class to YOLO: https://github.com/ultralytics/yolov5/issues/1071

[miguemorales]

ok, i will review that reference and uplad any new. Thanks