What are the parameters for training only some classes?

[WonTaeYeon]

What are the parameters for training only some classes?

Which of the ways does EfficinetDet improve accuracy, learning only specific classes in a pre-learned model or learning only specific classes?

Is EfficientDet suitable for working on real-time video? (8 fps or more guaranteed on CPU only computer)

[CraigWang1]

I am wondering this too, I tried adding this
--hparams="use_bfloat16=false,num_classes=1,learning_rate=0.001"

[mingxingtan]

Hi, you can finetine from COCO checkpoint and set "num_classes=1"? See https://github.com/google/automl/tree/master/efficientdet#7-finetune-on-pascal-voc-2012-with-detector-coco-ckpt

I am going to close this issue for now, but feel free to reopen it if you still have problems.