How to set the NUM_CLASSES, if I want to train only one class? #2708

Unanswered

AstrayChao asked this question in Q&A

AstrayChao
Mar 8, 2021

hi, I want to train a person detector, and the NUM_CLASSES should be 1 or 2 ('person' + 'background')?
thank you

Replies: 1 comment 1 reply

ppwwyyxx
Mar 9, 2021

see https://detectron2.readthedocs.io/en/latest/modules/config.html#config-references

1 reply

AstrayChao Mar 15, 2021
Author

thanks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment