How to set different levels of the mask in y_train_dir as mask[..., 0], mask[..., 1] ... ?

alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset

http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html

847 stars 517 forks source link

How to set different levels of the mask in y_train_dir as mask[..., 0], mask[..., 1] ... ? #166

Open jtyytim opened 2 years ago

jtyytim commented 2 years ago

Hello,

For the multi-class segmentation, I want to know how to set different levels of the mask in y_train_dir as mask[..., 0], mask[..., 1] ... like that? Can you explain how to realize it with my own data?

Thanks, Thiem