Skip to content

Commit f9a5c19

Browse files
committed
STYLE: use pythonic filters
1 parent 2ca20c1 commit f9a5c19

File tree

1 file changed

+3
-15
lines changed
  • src/Filtering/Smoothing/SmoothImageWithDiscreteGaussianFilter

1 file changed

+3
-15
lines changed

src/Filtering/Smoothing/SmoothImageWithDiscreteGaussianFilter/Code.py

Lines changed: 3 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -25,20 +25,8 @@
2525
parser.add_argument("sigma", type=float)
2626
args = parser.parse_args()
2727

28-
PixelType = itk.UC
29-
Dimension = 2
28+
image = itk.imread(args.input_image)
3029

31-
ImageType = itk.Image[PixelType, Dimension]
30+
smooth_image = itk.discrete_gaussian_image_filter(image, sigma=args.sigma)
3231

33-
reader = itk.ImageFileReader[ImageType].New()
34-
reader.SetFileName(args.input_image)
35-
36-
smoother = itk.DiscreteGaussianImageFilter[ImageType, ImageType].New()
37-
smoother.SetInput(reader.GetOutput())
38-
smoother.SetSigma(args.sigma)
39-
40-
writer = itk.ImageFileWriter[ImageType].New()
41-
writer.SetFileName(args.output_image)
42-
writer.SetInput(smoother.GetOutput())
43-
44-
writer.Update()
32+
itk.imwrite(smooth_image, args.output_image)

0 commit comments

Comments
 (0)