├── .gitignore ├── README.md ├── braindiffusion.egg-info ├── PKG-INFO ├── SOURCES.txt ├── dependency_links.txt └── top_level.txt ├── braindiffusion ├── __init__.py ├── __pycache__ │ └── __init__.cpython-37.pyc ├── codex_ae │ ├── __init__.py │ ├── auto_encoder.py │ ├── dewave_ae.py │ ├── dewave_codex_ae.py │ ├── freq_codex_ae.py │ ├── model_decoding.py │ ├── nearest_embed.py │ ├── temp.py │ ├── test_nearest_embed.py │ └── util.py ├── data │ ├── __init__.py │ ├── bci_comp_iv.py │ ├── zuco_nr.py │ └── zuco_nr_freq.py ├── modeling │ ├── __init__.py │ ├── eeg_spectro_encoder.py │ └── x_transformer.py ├── pipeline_freqmap_diffusion.py ├── pipeline_wave_diffusion.py ├── utils │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ ├── sampling_func.cpython-37.pyc │ │ └── wave_spectron.cpython-37.pyc │ ├── construct_wave_mat_to_pickle_v1.py │ ├── construct_wave_mat_to_pickle_v2.py │ ├── data_loading_helpers.py │ ├── mne_visualize.py │ ├── sampling_func.py │ ├── wave_spectron.py │ └── wave_tokenizer.py └── visualization │ ├── spectrogram.png │ └── test.png ├── config ├── accelerate_deepspeed.yaml ├── accelerate_local.yaml ├── accelerate_sagemaker.yaml └── ldm_autoencoder_kl.yaml ├── docs └── DATA.md ├── file_to_remove.txt ├── scripts ├── data_preparation_wave.sh ├── encode_text.py ├── freq2spectro_zuco.py ├── freq2spectromap8-105-56_zuco.py ├── train_raw_waves_unet.py ├── train_raw_waves_unet_conditioned.py ├── train_unet_freq.py ├── train_unet_freq_2d.py ├── waves2spectro.py └── waves2spectro_zuco.py ├── setup.py └── visualization ├── braindiffusion.png ├── clean_image0.png ├── freqimg ├── featuremap_images:99_bs0.png └── featuremap_images:9_bs0.png ├── merged_image.png ├── psd_mapwave_97.png ├── spectrogram.png ├── stft_spetrogram.png ├── topomap_images:1_0.gif ├── topomap_images:1_0.png └── vismel_bci_iv.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/README.md -------------------------------------------------------------------------------- /braindiffusion.egg-info/PKG-INFO: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion.egg-info/PKG-INFO -------------------------------------------------------------------------------- /braindiffusion.egg-info/SOURCES.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion.egg-info/SOURCES.txt -------------------------------------------------------------------------------- /braindiffusion.egg-info/dependency_links.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /braindiffusion.egg-info/top_level.txt: -------------------------------------------------------------------------------- 1 | braindiffusion 2 | -------------------------------------------------------------------------------- /braindiffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /braindiffusion/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /braindiffusion/codex_ae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /braindiffusion/codex_ae/auto_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/auto_encoder.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/dewave_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/dewave_ae.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/dewave_codex_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/dewave_codex_ae.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/freq_codex_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/freq_codex_ae.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/model_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/model_decoding.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/nearest_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/nearest_embed.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/temp.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /braindiffusion/codex_ae/test_nearest_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/test_nearest_embed.py -------------------------------------------------------------------------------- /braindiffusion/codex_ae/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/codex_ae/util.py -------------------------------------------------------------------------------- /braindiffusion/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /braindiffusion/data/bci_comp_iv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/data/bci_comp_iv.py -------------------------------------------------------------------------------- /braindiffusion/data/zuco_nr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/data/zuco_nr.py -------------------------------------------------------------------------------- /braindiffusion/data/zuco_nr_freq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/data/zuco_nr_freq.py -------------------------------------------------------------------------------- /braindiffusion/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /braindiffusion/modeling/eeg_spectro_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/modeling/eeg_spectro_encoder.py -------------------------------------------------------------------------------- /braindiffusion/modeling/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/modeling/x_transformer.py -------------------------------------------------------------------------------- /braindiffusion/pipeline_freqmap_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/pipeline_freqmap_diffusion.py -------------------------------------------------------------------------------- /braindiffusion/pipeline_wave_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/pipeline_wave_diffusion.py -------------------------------------------------------------------------------- /braindiffusion/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /braindiffusion/utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /braindiffusion/utils/__pycache__/sampling_func.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/__pycache__/sampling_func.cpython-37.pyc -------------------------------------------------------------------------------- /braindiffusion/utils/__pycache__/wave_spectron.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/__pycache__/wave_spectron.cpython-37.pyc -------------------------------------------------------------------------------- /braindiffusion/utils/construct_wave_mat_to_pickle_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/construct_wave_mat_to_pickle_v1.py -------------------------------------------------------------------------------- /braindiffusion/utils/construct_wave_mat_to_pickle_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/construct_wave_mat_to_pickle_v2.py -------------------------------------------------------------------------------- /braindiffusion/utils/data_loading_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/data_loading_helpers.py -------------------------------------------------------------------------------- /braindiffusion/utils/mne_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/mne_visualize.py -------------------------------------------------------------------------------- /braindiffusion/utils/sampling_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/sampling_func.py -------------------------------------------------------------------------------- /braindiffusion/utils/wave_spectron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/wave_spectron.py -------------------------------------------------------------------------------- /braindiffusion/utils/wave_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/utils/wave_tokenizer.py -------------------------------------------------------------------------------- /braindiffusion/visualization/spectrogram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/visualization/spectrogram.png -------------------------------------------------------------------------------- /braindiffusion/visualization/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/braindiffusion/visualization/test.png -------------------------------------------------------------------------------- /config/accelerate_deepspeed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/config/accelerate_deepspeed.yaml -------------------------------------------------------------------------------- /config/accelerate_local.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/config/accelerate_local.yaml -------------------------------------------------------------------------------- /config/accelerate_sagemaker.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/config/accelerate_sagemaker.yaml -------------------------------------------------------------------------------- /config/ldm_autoencoder_kl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/config/ldm_autoencoder_kl.yaml -------------------------------------------------------------------------------- /docs/DATA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/docs/DATA.md -------------------------------------------------------------------------------- /file_to_remove.txt: -------------------------------------------------------------------------------- 1 | ./backup 2 | ./models -------------------------------------------------------------------------------- /scripts/data_preparation_wave.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/data_preparation_wave.sh -------------------------------------------------------------------------------- /scripts/encode_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/encode_text.py -------------------------------------------------------------------------------- /scripts/freq2spectro_zuco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/freq2spectro_zuco.py -------------------------------------------------------------------------------- /scripts/freq2spectromap8-105-56_zuco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/freq2spectromap8-105-56_zuco.py -------------------------------------------------------------------------------- /scripts/train_raw_waves_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/train_raw_waves_unet.py -------------------------------------------------------------------------------- /scripts/train_raw_waves_unet_conditioned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/train_raw_waves_unet_conditioned.py -------------------------------------------------------------------------------- /scripts/train_unet_freq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/train_unet_freq.py -------------------------------------------------------------------------------- /scripts/train_unet_freq_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/train_unet_freq_2d.py -------------------------------------------------------------------------------- /scripts/waves2spectro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/waves2spectro.py -------------------------------------------------------------------------------- /scripts/waves2spectro_zuco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/scripts/waves2spectro_zuco.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/setup.py -------------------------------------------------------------------------------- /visualization/braindiffusion.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/braindiffusion.png -------------------------------------------------------------------------------- /visualization/clean_image0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/clean_image0.png -------------------------------------------------------------------------------- /visualization/freqimg/featuremap_images:99_bs0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/freqimg/featuremap_images:99_bs0.png -------------------------------------------------------------------------------- /visualization/freqimg/featuremap_images:9_bs0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/freqimg/featuremap_images:9_bs0.png -------------------------------------------------------------------------------- /visualization/merged_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/merged_image.png -------------------------------------------------------------------------------- /visualization/psd_mapwave_97.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/psd_mapwave_97.png -------------------------------------------------------------------------------- /visualization/spectrogram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/spectrogram.png -------------------------------------------------------------------------------- /visualization/stft_spetrogram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/stft_spetrogram.png -------------------------------------------------------------------------------- /visualization/topomap_images:1_0.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/topomap_images:1_0.gif -------------------------------------------------------------------------------- /visualization/topomap_images:1_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/topomap_images:1_0.png -------------------------------------------------------------------------------- /visualization/vismel_bci_iv.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/duanyiqun/Brain-Diffusion/HEAD/visualization/vismel_bci_iv.png --------------------------------------------------------------------------------