From d68c76b4fd9e8341756810f04219066c0e60757b Mon Sep 17 00:00:00 2001 From: shenyan <23357320@qq.com> Date: Fri, 15 Oct 2021 19:26:18 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9visual=5Flabel=E4=B8=AD?= =?UTF-8?q?=E5=88=9B=E5=BB=BA=E5=A4=9A=E7=BA=A7=E7=9B=AE=E5=BD=95=E7=9A=84?= =?UTF-8?q?BUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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])