Build the directory label
KennyChen880127 opened this issue · 0 comments
KennyChen880127 commented
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.
As far as I know I need to edit data.py class DirectoryDataset(Dataset) ,Am I right?
Hope someon can help me to explain.. Thanks a lot!