Browse Source

Update get_hed.py

pull/3/head
troyyyyy GitHub 2 years ago
parent
commit
7fbe96fa79
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      datasets/hed/get_hed.py

+ 1
- 1
datasets/hed/get_hed.py View File

@@ -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:


Loading…
Cancel
Save