Create dataset_loader.py
Browse files- 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 |
+
|