mhamilton723 / STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences
MIT License
711 stars 142 forks source link

Build the directory label #73

Closed KennyChen880127 closed 1 year ago

KennyChen880127 commented 1 year ago

Hello everyone! I'm training my onw dataset. But I have no idea where can I set the directory label likes train_Segmentation.py get_class_labels, maybe like car, bike... And how to setting the dir_dataset_n_classes to correspond the label.

Hope someon can help me to explain.. Thanks a lot!