Skip to content

Commit f8f5e6f

Browse files
committed
Fix noise addition parameter classes (ultimate fix!)
1 parent c6bab15 commit f8f5e6f

2 files changed

Lines changed: 12 additions & 24 deletions

File tree

datalab/gui/processor/image.py

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@
2121
from sigima.objects import (
2222
ImageObj,
2323
ImageROI,
24-
NormalDistribution2DParam,
25-
PoissonDistribution2DParam,
24+
NormalDistributionParam,
25+
PoissonDistributionParam,
2626
ROI2DParam,
27-
UniformDistribution2DParam,
27+
UniformDistributionParam,
2828
)
2929
from sigima.objects.scalar import GeometryResult
3030
from sigima.proc.decorator import ComputationMetadata
@@ -422,19 +422,13 @@ def register_processing(self) -> None:
422422
)
423423
# Noise addition
424424
self.register_1_to_1(
425-
sipi.add_gaussian_noise,
426-
_("Add Gaussian noise"),
427-
NormalDistribution2DParam,
425+
sipi.add_gaussian_noise, _("Add Gaussian noise"), NormalDistributionParam
428426
)
429427
self.register_1_to_1(
430-
sipi.add_poisson_noise,
431-
_("Add Poisson noise"),
432-
PoissonDistribution2DParam,
428+
sipi.add_poisson_noise, _("Add Poisson noise"), PoissonDistributionParam
433429
)
434430
self.register_1_to_1(
435-
sipi.add_uniform_noise,
436-
_("Add uniform noise"),
437-
UniformDistribution2DParam,
431+
sipi.add_uniform_noise, _("Add uniform noise"), UniformDistributionParam
438432
)
439433
# Noise reduction
440434
self.register_1_to_1(

datalab/gui/processor/signal.py

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,12 @@
1717
import sigima.proc.signal as sips
1818
from guidata.qthelpers import exec_dialog
1919
from sigima.objects import (
20-
NormalDistribution1DParam,
21-
PoissonDistribution1DParam,
20+
NormalDistributionParam,
21+
PoissonDistributionParam,
2222
ROI1DParam,
2323
SignalObj,
2424
SignalROI,
25-
UniformDistribution1DParam,
25+
UniformDistributionParam,
2626
create_signal,
2727
)
2828
from sigima.objects.scalar import GeometryResult, TableResult
@@ -231,19 +231,13 @@ def register_processing(self) -> None:
231231
)
232232
# Noise addition
233233
self.register_1_to_1(
234-
sips.add_gaussian_noise,
235-
_("Add Gaussian noise"),
236-
NormalDistribution1DParam,
234+
sips.add_gaussian_noise, _("Add Gaussian noise"), NormalDistributionParam
237235
)
238236
self.register_1_to_1(
239-
sips.add_poisson_noise,
240-
_("Add Poisson noise"),
241-
PoissonDistribution1DParam,
237+
sips.add_poisson_noise, _("Add Poisson noise"), PoissonDistributionParam
242238
)
243239
self.register_1_to_1(
244-
sips.add_uniform_noise,
245-
_("Add uniform noise"),
246-
UniformDistribution1DParam,
240+
sips.add_uniform_noise, _("Add uniform noise"), UniformDistributionParam
247241
)
248242
# Noise reduction
249243
self.register_1_to_1(

0 commit comments

Comments
 (0)