NVIDIA / DIGITS

Deep Learning GPU Training System
https://developer.nvidia.com/digits
BSD 3-Clause "New" or "Revised" License
4.11k stars 1.38k forks source link

Missing labels file in downloaded semantic segmentation networks #1360

Open skycoop opened 7 years ago

skycoop commented 7 years ago

Since segmentation networks and datasets are stored in the generic form, the downloaded models don't include a labels file. DIGITS does use the labels file provided at dataset creation for the visualizations, so it definitely keeps track of the labels. It would be extremely helpful to have that included in the downloaded archive since the semantic segmentation use case requires some way to convert numeric labels to text ones.

gheinrich commented 7 years ago

Thanks. Tracking this as an enhancement.