scikit-image

1
# See "Writing benchmarks" in the asv docs for more information.
2
# https://asv.readthedocs.io/en/latest/writing_benchmarks.html
3
import numpy as np
4
from skimage import util
5

6

7
class NoiseSuite:
8
    """Benchmark for noise routines in scikit-image."""
9

10
    params = ([0.0, 0.50, 1.0], [0.0, 0.50, 1.0])
11

12
    def setup(self, *_):
13
        self.image = np.zeros((5000, 5000))
14

15
    def peakmem_salt_and_pepper(self, amount, salt_vs_pepper):
16
        self._make_salt_and_pepper_noise(amount, salt_vs_pepper)
17

18
    def time_salt_and_pepper(self, amount, salt_vs_pepper):
19
        self._make_salt_and_pepper_noise(amount, salt_vs_pepper)
20

21
    def _make_salt_and_pepper_noise(self, amount, salt_vs_pepper):
22
        util.random_noise(
23
            self.image,
24
            mode="s&p",
25
            amount=amount,
26
            salt_vs_pepper=salt_vs_pepper,
27
        )
28