This repository contains the code used to analyse the submissions for the Inaugural Flatiron Cryo-EM Heterogeneity Challenge.
This repository explains how to preprocess a submission (80 maps and corresponding probability distribution), and analyze it. Challenge participants can benchmark their submissions locally against the ground truth and other submissions that are available on the cloud via the Open Science Foundation project The Inaugural Flatiron Institute Cryo-EM Heterogeneity Community Challenge.
This is a work in progress, while the code will probably not change, we are still writting better tutorials, documentation, and other ideas for analyzing the data. We are also in the process of making it easier for other people to contribute with their own metrics and methods. We are also in the process of distributing the code to PyPi.
The data is available via the Open Science Foundation project The Inaugural Flatiron Institute Cryo-EM Heterogeneity Community Challenge. You can download via a web browser, or programatically with wget as per this script.
NOTE: We recommend downloadaing the data with the script and wget as the downloads from the web browser might be unstable.
Installing this repository is simply. We recommend creating a virtual environment (using conda or pyenv), since we have dependencies such as PyTorch or Aspire, which are better dealt with in an isolated environment. After creating your environment, make sure to activate it and run
cd /path/to/Cryo-EM-Heterogeneity-Challenge-1
pip install .If you are interested in testing the programs previously installed, please, install the repository in development mode with the following commands:
cd /path/to/Cryo-EM-Heterogeneity-Challenge-1
pip install .[dev]The test included in the repo can be executed with PyTest as shown below:
cd /path/to/Cryo-EM-Heterogeneity-Challenge-1
sh tests/scripts/fetch_test_data.sh # download test data from OSF
pytest tests/test_preprocessing.py
pytest tests/test_svd.py
pytest tests/test_map_to_map.py
pytest tests/test_distribution_to_distribution.pyIf you want to run our code on the full challenge data, or you own local data, please complete the following steps
1. Download the full challenge data from The Inaugural Flatiron Institute Cryo-EM Heterogeneity Community Challenge
You can do this through the web browser, or programatically with wget (you can use this script, this will download around 220 GB of data)
Point to the path where the data is locally The tutorial notebooks explain how to setup the config files, and run the commands
cryo_challenge run_preprocessing --config config_files/config_preproc.yaml
cryo_challenge run_svd --config config_files/config_svd.yaml
cryo_challenge run_map2map_pipeline --config config_files/config_map_to_map.yaml
cryo_challenge run_distribution2distribution_pipeline --config config_files/config_distribution_to_distribution.yaml
If you find any bug or have a suggestion on the code feel free to open an issue here.
We also welcome any help with the development of this repository. If you want to contribute with your own suggestions, code, or fixes, we recommend creating a fork of this repository to avoid any incompatibilities with newer versions of the software. Once you are happy with your new code, please, make a PR from your fork to this repository.
We are also working on pipelines to simplify the exentension of the code with new metrics or functionalities, stay tuned!
- Miro A. Astore, Geoffrey Woollard, David Silva-Sánchez, Wenda Zhao, Khanh Dao Duc, Nikolaus Grigorieff, Pilar Cossio, and Sonya M. Hanson. "The Inaugural Flatiron Institute Cryo-EM Heterogeneity Community Challenge". 9 June 2023. DOI:10.17605/OSF.IO/8H6FZ
- David Herreros for testing and CI and debugging in this repo