From 7fbe96fa79c1aef602edb5b167fdaa19c659e5d3 Mon Sep 17 00:00:00 2001
From: troyyyyy <49091847+troyyyyy@users.noreply.github.com>
Date: Wed, 22 Feb 2023 17:39:53 +0800
Subject: [PATCH] Update get_hed.py

---
 datasets/hed/get_hed.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datasets/hed/get_hed.py b/datasets/hed/get_hed.py
index 8b61b62..b317d94 100644
--- a/datasets/hed/get_hed.py
+++ b/datasets/hed/get_hed.py
@@ -41,7 +41,7 @@ def get_pretrain_data(labels, image_size=(28, 28, 1)):
     X = []
     for label in labels:
         label_path = os.path.join(
-            "./datasets/hed/dataset/mnist_images", label
+            "./datasets/hed/mnist_images", label
         )
         img_path_list = os.listdir(label_path)
         for img_path in img_path_list: