diff --git a/utils.py b/utils.py index 41e8499..e34a8de 100644 --- a/utils.py +++ b/utils.py @@ -87,7 +87,7 @@ def visual_label(dataset_path, n_classes): from torchvision import transforms trans_factory = transforms.ToPILImage() if not os.path.exists(dataset_path + '/visual_label'): - os.mkdir(dataset_path + '/visual_label') + os.makedirs(dataset_path + '/visual_label') for index in range(len(label_image_list)): label_image = cv2.imread(label_image_list[index], -1) name = os.path.basename(label_image_list[index])