Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
[deeplab] Add Cityscapes and ADE20K to the notebook demo #6991
The current demo only includes an example for using the DeepLab model trained on the Pascal dataset.
Judging from the issues, this is a common issue, and since all of the required infra seems to be there, adding support for Cityscapes and ADE20K will serve as a good first PR for the community newcomers and help eliminate the confusion around different labelling schemes for those who see the demo first. Personally, this bit me while porting DeepLab to
cityscapes_labels = [ 'road', 'sidewalk', 'building', 'wall', 'fence', 'pole', 'traffic light', 'traffic sign', 'vegetation', 'terrain', 'sky', 'person', 'rider', 'car', 'truck', 'bus', 'train', 'motorcycle', 'bicycle', ]
Do I understand correctly that the frozen model trained on Cityscapes produces labels according to the list above?
UPDATE 11/06: I can confirm that it does. See the segmentation demo below:
Thank you for your post. We noticed you have not filled out the following field in the issue template. Could you update them if they are relevant in your case, or leave them as N/A? Thanks.