tensorflow-deeplab-v3-plus icon indicating copy to clipboard operation
tensorflow-deeplab-v3-plus copied to clipboard

Published 20 hours ago • verified publisher icon rishizek

Training in two classes

Open sonfire186 opened this issue 5 years ago • 2 comments

I need to train the network only in two classes from PASLAC VOC. Background and personal.

train.py

_NUM_CLASSES = 2

preprocessing.py

label_colours = [(0, 0, 0),  # 0=background
                 (192, 128, 128) # 15=person]

Am I doing the right thing?

sonfire186 avatar Feb 26 '19 08:02 sonfire186

Yes, if your training set, person label, is 192,128,128, that should work. I use (128,0,0) to annotate the other class

northeastsquare avatar Apr 08 '19 10:04 northeastsquare

what should I set If I want to retrain the model in two classes， that is， I want to the model output two classes

DomicZhong avatar Apr 28 '20 12:04 DomicZhong