Specifying one class to be segmented

[k3nnywilliam]

Hi @gustavz

Really cool work! Is there a way to just segment one class, i.e., just a person, instead of detecting various objects in the scene?

[NguyenVanThanhHust]

You can look at his code. This part : LABEL_NAMES = np.asarray([
'background', 'aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus',
'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike',
'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tv'
])

You can choose to display result only if your class is person.