search

bbaaii / DreamDiffusion

Implementation of “DreamDiffusion: Generating High-Quality Images from Brain EEG Signals”
MIT License
429 stars 49 forks source link

.npy Extension Issue #12

Open aryonna-rice opened 8 months ago

aryonna-rice commented 8 months ago

I am trying to train the model using default parameters. All of the files I downloaded from "eeg" on the readme end in .pth even though there are lines of code that indicate that a .npy file is expected. What can I do to fix this issue? Is there something I am missing? Additionally, the datasets/mne_data folder does not exist for me. Where was I supposed to get this from? I am trying to use this framework for a school project so any insight would be greatly appreciated.

bbaaii commented 8 months ago

You can directly download this data from the MOABB project https://github.com/NeuroTechX/moabb and use their respective processing code for operations like filtering. Finally, save them as npy files.

zhast commented 7 months ago

Could you be more specific as to which files from the MOABB repo to save as .npy in the datasets/mne_data folder?

taziksh commented 5 months ago

@bbaaii was there a specific set of MOABB datasets you used for pre-training? Or does it not matter because it's just auto encoder?