Skip to content
#

segmentation-models

Here are 8 public repositories matching this topic...

captainst
captainst commented Oct 15, 2019

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

Improve this page

Add a description, image, and links to the segmentation-models topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the segmentation-models topic, visit your repo's landing page and select "manage topics."

Learn more

You can’t perform that action at this time.