alignvis
收藏资源简介:
这个数据集包含了图像、图像嵌入和脑电图数据,用于研究和应用。图像和嵌入数据以tar文件形式提供,可以通过WebDataset库进行加载。脑电图数据以numpy数组的形式存储,并且提供了如何从Hugging Face Hub中获取这些数据的说明。此外,README文件中还包含了如何将数据集下载到本地,并从tar文件中提取数据以供本地使用的说明。
This dataset comprises images, image embeddings, and electroencephalogram (EEG) data intended for research and practical applications. The images and image embedding data are distributed as tar archives, which can be loaded using the WebDataset library. The EEG data is stored as NumPy arrays, and instructions on how to obtain these data from the Hugging Face Hub are provided. Furthermore, the README file includes guidance on downloading the dataset to a local environment and extracting data from the tar files for local use.
数据集概述
基本信息
- 数据集名称: alignvis
- 存储库ID: nonarjb/alignvis
- 数据集类型: WebDataset格式的分片数据集
- 主要数据目录: things_eeg_2
数据内容
数据分片类型
- 图像数据:
things_eeg_2-images-*.tar- 包含图像文件 - 图像嵌入向量:
things_eeg_2-image_embeddings-*.tar- 包含向量嵌入数据(.npy/.npz格式) - 预处理EEG数据:
things_eeg_2-preprocessed_eeg-*.tar- 包含EEG数组数据(.npy/.npz格式)
文件组织结构
每个分片中的WebDataset __key__对应文件在顶层文件夹下的相对路径(不含扩展名)。原始相对路径可通过以下方式重构:
rel_path = "<top>/" + key + "." + <ext>
数据加载方式
图像数据加载
- 支持通过PIL库加载原始图像
- 保留原始相对路径信息
- 返回RGB格式的图像数据
图像嵌入向量加载
- 支持加载.npy和.npz格式的嵌入向量
- 自动转换为torch.Tensor格式
- 支持多种字段名称的嵌入向量(embedding, emb, vector, feat, features, clip, image, text)
EEG数据加载
- 支持加载预处理后的EEG数据
- 按受试者ID和训练/测试划分加载数据
- 返回EEG数据数组和通道名称信息
本地使用
支持下载数据集并解压到本地目录,保持原始文件结构:
- 图像数据:
./restore/things_eeg_2/images/ - 图像嵌入:
./restore/things_eeg_2/image_embeddings/ - EEG数据:
./restore/things_eeg_2/preprocessed_eeg/
相关数据集
- things_meg: 可通过替换
dataset_dir="things_eeg_2"为dataset_dir="things_meg"使用
依赖要求
- webdataset
- huggingface_hub
- pillow
- torch
- tqdm
- 可选: hf_transfer(用于大文件传输加速)




