lowbrightnessgenerator.py 文件源码

def fit(self, noise_samples, track_samples):
    import scipy.stats as stats

    track_area = np.sum(track_samples > 0, axis=(1, 2))
    area_distribution_params = stats.expon.fit(track_area)

    area_distribution = stats.expon(*area_distribution_params)

    return LowBrightnessGenerator(
      noise_samples, track_samples,
      signal_level=self._signal_level,
      track_rate=self._track_rate,
      white_noise_rate=self._white_noise_rate,
      white_noise_maximum=self._white_noise_maximum,
      pseudo_tracks_rate=self._pseudo_tracks_rate,
      pseudo_track_sparseness=self._pseudo_track_sparseness,
      pseudo_track_width=self._pseudo_track_width,
      area_distribution=area_distribution
    )