Skip to content

Commit

Permalink
change background_median to background_mean
Browse files Browse the repository at this point in the history
  • Loading branch information
randlet committed Mar 31, 2021
1 parent f069364 commit 41b72a7
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions pylinac/ct.py
Expand Up @@ -58,7 +58,7 @@ class HUDiskROI(DiskROI):
def __init__(self, array: np.ndarray, angle: float, roi_radius: float, dist_from_center: float,
phantom_center: Union[tuple, Point], nominal_value: Optional[int] = None,
tolerance: Optional[int] = None,
background_median: Optional[float]=None, background_std: Optional[float]=None):
background_mean: Optional[float]=None, background_std: Optional[float]=None):
"""
Parameters
----------
Expand All @@ -70,15 +70,15 @@ def __init__(self, array: np.ndarray, angle: float, roi_radius: float, dist_from
super().__init__(array, angle, roi_radius, dist_from_center, phantom_center)
self.nominal_val = nominal_value
self.tolerance = tolerance
self.background_median = background_median
self.background_mean = background_mean
self.background_std = background_std

@property
def cnr(self) -> float:
"""The contrast-to-noise value of the HU disk"""
if self.background_median is None or self.background_std is None:
if self.background_mean is None or self.background_std is None:
return None
return 2*abs(self.pixel_value - self.background_median) / (self.std + self.background_std)
return 2*abs(self.pixel_value - self.background_mean) / (self.std + self.background_std)

@property
def value_diff(self) -> float:
Expand Down Expand Up @@ -251,17 +251,17 @@ def _setup_rois(self) -> None:
self.background_rois[name] = HUDiskROI(self.image, setting['angle_corrected'], setting['radius_pixels'], setting['distance_pixels'],
self.phan_center)
if self.background_rois:
background_median = np.mean([roi.pixel_value for roi in self.background_rois.values()])
background_mean = np.mean([roi.pixel_value for roi in self.background_rois.values()])
background_std = np.std([roi.pixel_value for roi in self.background_rois.values()])
else:
background_median = None
background_mean = None
background_std = None

for name, setting in self.roi_settings.items():
nominal_value = setting.get('value', 0)
self.rois[name] = HUDiskROI(self.image, setting['angle_corrected'], setting['radius_pixels'], setting['distance_pixels'],
self.phan_center, nominal_value, self.tolerance,
background_median=background_median, background_std=background_std)
background_mean=background_mean, background_std=background_std)

# TODO: better define threshold
def plot_rois(self, axis: plt.Axes, threshold=None) -> None:
Expand Down

0 comments on commit 41b72a7

Please sign in to comment.