Skip to content

match_template_peaks

Data class and helper functions for match template peaks and statistics.

MatchTemplatePeaks

Bases: NamedTuple

Helper class for return value of extract_peaks_and_statistics.

Source code in src/leopard_em/analysis/match_template_peaks.py
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
class MatchTemplatePeaks(NamedTuple):
    """Helper class for return value of extract_peaks_and_statistics."""

    pos_y: torch.Tensor
    pos_x: torch.Tensor
    mip: torch.Tensor
    scaled_mip: torch.Tensor
    psi: torch.Tensor
    theta: torch.Tensor
    phi: torch.Tensor
    relative_defocus: torch.Tensor
    correlation_mean: torch.Tensor
    correlation_variance: torch.Tensor
    total_correlations: int

match_template_peaks_to_dataframe(peaks)

Convert MatchTemplatePeaks object to a pandas DataFrame.

Source code in src/leopard_em/analysis/match_template_peaks.py
30
31
32
def match_template_peaks_to_dataframe(peaks: MatchTemplatePeaks) -> pd.DataFrame:
    """Convert MatchTemplatePeaks object to a pandas DataFrame."""
    return pd.DataFrame(peaks._asdict())

match_template_peaks_to_dict(peaks)

Convert MatchTemplatePeaks object to a dictionary.

Source code in src/leopard_em/analysis/match_template_peaks.py
25
26
27
def match_template_peaks_to_dict(peaks: MatchTemplatePeaks) -> dict:
    """Convert MatchTemplatePeaks object to a dictionary."""
    return peaks._asdict()