Skip to content

Cl mask patch #1171

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Sep 4, 2024
Merged

Cl mask patch #1171

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b274690
fix fuzzy mask dtypes
garrettwrong Sep 3, 2024
9c76758
fix rise time
garrettwrong Sep 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion src/aspire/abinitio/commonline_base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import numpy as np
import scipy.sparse as sparse

from aspire.image import Image
from aspire.operators import PolarFT
from aspire.utils import common_line_from_rots, fuzzy_mask, tqdm
from aspire.utils.random import choice
Expand Down Expand Up @@ -91,8 +92,14 @@ def _prepare_pf(self):
imgs = self.src.images[:]

if self.mask:
fuzz_mask = fuzzy_mask((self.n_res, self.n_res), self.dtype)
# For best results and to reproduce MATLAB:
# Set risetime=2
# Always compute mask (erf) in doubles.
fuzz_mask = fuzzy_mask((self.n_res, self.n_res), np.float64, risetime=2)
# Apply mask in doubles (allow imgs to upcast as needed)
imgs = imgs * fuzz_mask
# Cast to desired type
imgs = Image(imgs.asnumpy().astype(self.dtype, copy=False))

# Obtain coefficients of polar Fourier transform for input 2D images
pft = PolarFT(
Expand Down