Open
Description
In file binary segmentation (camvid).ipynb, block 5, there is:
# Lets look at data we have
dataset = Dataset(x_train_dir, y_train_dir, classes=['car', 'pedestrian'])
image, mask = dataset[5] # get some sample
visualize(
image=image,
cars_mask=mask[..., 0].squeeze(),
sky_mask=mask[..., 1].squeeze(),
background_mask=mask[..., 2].squeeze(),
)
here, sky_mask should be pedestrain. I was first a bit confused to see the title on plot shown as "sky_mask" while the mask shows pedestrain.
Excellant work man and many thanks !
Metadata
Metadata
Assignees
Labels
No labels