How to extend category at Object Detector model without Relabeling?

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Object detectors are typically learned based on fully-annotated training data with fixed pre-defined categories,In such scenario, only the original training set annotated with the old classes and some new training data labeled with the new classes are available.how to extend category?

2

According to your comment above, the new classes have label, right?

3

yeah, new training data labeled with the new classes

4

So, why you ask “How to extend category at Object Detector model without Relabeling?”

5

the old data have not new category label, the new dataset have not old category label,it can add new category on old tlt model?

6

OK, I am afraid you want to add new dataset with new classes. Yes, it is possible. Please modify the training spec and then run training.

7

how to modify?
add new target_class_mapping in the training spec and then run training with new dataset & old tlt model?

8

Yes, adding new target_class_mapping in the training spec.
You can also use the old tlt model as the pretrained model.
BTW, which network did you train? Detectnet_v2 or yolo_v4?

9

yolo_v4

10

Training data contains new dataset and old dataset? or only new dataset ??

11

new dataset and old dataset.

12

The new dataset only have new classes and old dataset only have old classes,Will it affect each other？

13

No.