drixo commited on
Commit
c530719
·
verified ·
1 Parent(s): 400eeef

Create dataset_loader.py

Browse files
Files changed (1) hide show
  1. dataset_loader.py +16 -0
dataset_loader.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from PIL import Image
3
+
4
+ def load_dataset(dataset_path="asl_dataset"):
5
+ mapping = {}
6
+ for letter in sorted(os.listdir(dataset_path)):
7
+ letter_path = os.path.join(dataset_path, letter)
8
+ if not os.path.isdir(letter_path):
9
+ continue
10
+ for img_file in os.listdir(letter_path):
11
+ img_path = os.path.join(letter_path, img_file)
12
+ mapping[img_file] = letter
13
+ return mapping
14
+
15
+ image_labels = load_dataset()
16
+