<jupyter_start><jupyter_text>-->-->- DatasetFolder```directory/├── class_x│   ├── xxx.ext│   ├── xxy.ext│   └── ...│       └── xxz.ext└── class_y    ├── 123.ext    ├── nsdf3.ext    └── ...    └── asd932_.ext```- ImageFolder```root/dog/xxx.pngroot/dog/xxy.pngroot/dog/[...]/xxz.pngroot/cat/123.pngroot/cat/nsdf3.pngroot/cat/[...]/asd932_.png```<jupyter_code>import os

os.chdir("../../")
from torch.utils.data import DataLoader
from gradsflow import data
from gradsflow.data.image import (
    RayImageFolder,
    image_dataset_from_directory,
    get_augmentations,
)
from torchvision import transforms as T
transforms = get_augmentations()


ds = RayImageFolder(
    "/Users/aniket/Pictures/data/train/",
    transforms,
)
iter_ds = iter(ds)
T.ToPILImage()(next(iter_ds)[0])<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.19254658385093168, 'fraction_numerical': 0.018633540372670808, 'mean_word_length': 3.7976190476190474, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 7, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '1', 'dir': 'jupyter-structured-clean-dedup', 'id': '170867', 'license': 'Apache-2.0', 'n_tokens_mistral': 328, 'n_tokens_neox': 323, 'n_words': 51, 'path': 'examples/nbs/04-RayDataset.ipynb', 'repo_name': 'stjordanis/gradsflow', 'size': '805'}