This is an algorithm for aligning three-dimensional objects represented as three dimensional arrays
The algorithm requires the following packages:
- ASPIRE
- numpy
- scipy
- mrcfile
- pymanopt
The user can simply download the files utils_BO.py and wemd.py. The main alignment algorithm can be found in utils_BO.py as align_BO, which takes in the two volumes
- Loss type: 'wemd' or 'eu'
- Downsampling level: an integer
- Number of iterations: an integer
- Whether to perform refinement: True or False
The data folder contains one test volume (more can be found on EMDB). The test_BOTalign.py conducts a comparison on four combinations of the parameters that we illustrated in the manuscript.
To reproduce the comparison results in the manuscript, install the EMalign and the alignOT packages, and run test_comparison.py. For alignOT, one needs to download all the files in their alignOT folder and put them in the same directory as test_comparison.py.