Background as a class in Semantic Segmentation #4399
Unanswered
Aleksandar1932
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I am doing binary semantic segmentation where I have one class that needs to be detected and the rest is background. Should background be considered as a class, hence setting the
MODEL.SEM_SEG_HEAD.NUM_CLASSES = 2
, or it should be set to 1.The training is done on a custom dataset which is registered according to the detectron2 documentation, and each sem_seg_file is a binary mask.
Beta Was this translation helpful? Give feedback.
All reactions