LSST Applications g0f08755f38+c89d42e150,g1635faa6d4+b6cf076a36,g1653933729+a8ce1bb630,g1a0ca8cf93+4c08b13bf7,g28da252d5a+f33f8200ef,g29321ee8c0+0187be18b1,g2bbee38e9b+9634bc57db,g2bc492864f+9634bc57db,g2cdde0e794+c2c89b37c4,g3156d2b45e+41e33cbcdc,g347aa1857d+9634bc57db,g35bb328faa+a8ce1bb630,g3a166c0a6a+9634bc57db,g3e281a1b8c+9f2c4e2fc3,g414038480c+077ccc18e7,g41af890bb2+e740673f1a,g5fbc88fb19+17cd334064,g7642f7d749+c89d42e150,g781aacb6e4+a8ce1bb630,g80478fca09+f8b2ab54e1,g82479be7b0+e2bd23ab8b,g858d7b2824+c89d42e150,g9125e01d80+a8ce1bb630,g9726552aa6+10f999ec6a,ga5288a1d22+065360aec4,gacf8899fa4+9553554aa7,gae0086650b+a8ce1bb630,gb58c049af0+d64f4d3760,gbd46683f8f+ac57cbb13d,gc28159a63d+9634bc57db,gcf0d15dbbd+e37acf7834,gda3e153d99+c89d42e150,gda6a2b7d83+e37acf7834,gdaeeff99f8+1711a396fd,ge2409df99d+cb1e6652d6,ge79ae78c31+9634bc57db,gf0baf85859+147a0692ba,gf3967379c6+02b11634a5,w.2024.45
LSST Data Management Base Package
Loading...
Searching...
No Matches
free_form.py
Go to the documentation of this file.
1# This file is part of scarlet_lite.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (https://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This program is free software: you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation, either version 3 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program. If not, see <https://www.gnu.org/licenses/>.
21
22__all__ = ["FreeFormComponent"]
23
24from typing import cast
25
26import numpy as np
27
28from ..bbox import Box
29from ..component import FactorizedComponent
30from ..detect import footprints_to_image
31from ..parameters import Parameter
32
33
35 """Implements a free-form component
36
37 With no constraints this component is typically either a garbage collector,
38 or part of a set of components to deconvolve an image by separating out
39 the different spectral components.
40
41 See `FactorizedComponent` for a list of parameters not shown here.
42
43 Parameters
44 ----------
45 peaks: `list` of `tuple`
46 A set of ``(cy, cx)`` peaks for detected sources.
47 If peak is not ``None`` then only pixels in the same "footprint"
48 as one of the peaks are included in the morphology.
49 If `peaks` is ``None`` then there is no constraint applied.
50 min_area: float
51 The minimum area for a peak.
52 If `min_area` is not `None` then all regions of the morphology
53 with fewer than `min_area` connected pixels are removed.
54 """
55
57 self,
58 bands: tuple,
59 spectrum: np.ndarray | Parameter,
60 morph: np.ndarray | Parameter,
61 model_bbox: Box,
62 bg_thresh: float | None = None,
63 bg_rms: np.ndarray | None = None,
64 floor: float = 1e-20,
65 peaks: list[tuple[int, int]] | None = None,
66 min_area: float = 0,
67 ):
68 super().__init__(
69 bands=bands,
70 spectrum=spectrum,
71 morph=morph,
72 bbox=model_bbox,
73 peak=None,
74 bg_rms=bg_rms,
75 bg_thresh=bg_thresh,
76 floor=floor,
77 )
78
79 self.peaks = peaks
80 self.min_area = min_area
81
82 def prox_spectrum(self, spectrum: np.ndarray) -> np.ndarray:
83 """Apply a prox-like update to the spectrum
84
85 This differs from `FactorizedComponent` because an
86 `SedComponent` has the spectrum normalized to unity.
87 """
88 # prevent divergent spectrum
89 spectrum[spectrum < self.floor] = self.floor
90 # Normalize the spectrum
91 spectrum = spectrum / np.sum(spectrum)
92 return spectrum
93
94 def prox_morph(self, morph: np.ndarray) -> np.ndarray:
95 """Apply a prox-like update to the morphology
96
97 This is the main difference between an `SedComponent` and a
98 `FactorizedComponent`, since this component has fewer constraints.
99 """
100 from lsst.scarlet.lite.detect_pybind11 import get_connected_multipeak, get_footprints # type: ignore
101
102 if self.bg_thresh is not None and isinstance(self.bg_rmsbg_rms, np.ndarray):
103 bg_thresh = self.bg_rmsbg_rms * self.bg_thresh
104 # Enforce background thresholding
105 model = self.spectrum[:, None, None] * morph[None, :, :]
106 morph[np.all(model < bg_thresh[:, None, None], axis=0)] = 0
107 else:
108 # enforce positivity
109 morph[morph < 0] = 0
110
111 if self.peaks is not None:
112 morph = morph * get_connected_multipeak(morph > 0, self.peaks, 0)
113
114 if self.min_area > 0:
115 footprints = get_footprints(morph > 0, 4.0, self.min_area, 0, False)
116 footprint_image = footprints_to_image(footprints, cast(tuple[int, int], morph.shape))
117 morph = morph * (footprint_image > 0).data
118
119 if np.all(morph == 0):
120 morph[0, 0] = self.floor
121
122 return morph
123
124 def resize(self, model_box: Box) -> bool:
125 return False
126
127 def __str__(self):
128 return (
129 f"FreeFormComponent(\n bands={self.bands}\n "
130 f"spectrum={self.spectrum})\n center={self.peak}\n "
131 f"morph_shape={self.morph.shape}"
132 )
133
134 def __repr__(self):
135 return self.__str____str__()
np.ndarray prox_morph(self, np.ndarray morph)
Definition free_form.py:94
np.ndarray prox_spectrum(self, np.ndarray spectrum)
Definition free_form.py:82
__init__(self, tuple bands, np.ndarray|Parameter spectrum, np.ndarray|Parameter morph, Box model_bbox, float|None bg_thresh=None, np.ndarray|None bg_rms=None, float floor=1e-20, list[tuple[int, int]]|None peaks=None, float min_area=0)
Definition free_form.py:67
MatrixB get_connected_multipeak(Eigen::Ref< const M > image, const std::vector< std::vector< int > > centers, const double thresh=0)
Proximal operator to trim pixels not connected to one of the source centers.
std::vector< Footprint > get_footprints(Eigen::Ref< const M > image, const double min_separation, const int min_area, const double thresh, const bool find_peaks=true)