File: itk_test_example.py

package info (click to toggle)
vistrails 2.1.1-1
  • links: PTS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 74,208 kB
  • ctags: 46,250
  • sloc: python: 316,267; xml: 52,512; sql: 3,627; php: 731; sh: 260; makefile: 108
file content (42 lines) | stat: -rw-r--r-- 1,151 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#example code to be pasted into a python terminal to test itk filters
#generally used to instaniate filters inorder lookup variable set methods

import itk
#image setup
dim = 2
pixeltype = itk.UC
kernel = itk.stle(dim, 5)
filename = "/home/mates/Desktop/brains.jpg"
reader = itk.ImageFileReader[itk.Image[pixeltype, dim]].New()
reader.SetFileName(filename)
reader.Update()

im = reader.GetOutput()
inImg = itk.Image[pixeltype, dim]
outImg = inImg

size_ = itk.Size[dim]()
size_.SetElement(0,150)
size_.SetElement(1,150)
region_ = itk.ImageRegion[dim]()
region_.SetSize(size_)

#filter
filt = itk.IsolatedConnectedImageFilter[inImg, outImg].New(im)
filt.SetReplaceValue(120)
filt.SetLower(155)
filt.SetUpperValueLimit(235)
filt.SetSeed1([120,120])
filt.SetSeed2([140,140])
filt.Update()

filt2 = itk.ConnectedThresholdImageFilter[itk.Image[pixeltype, dim], itk.Image[pixeltype, dim]].New(im)
filt2.SetSeed([100,100])
filt2.SetReplaceValue(100)
filt2.SetLower(150)
filt2.SetUpper(250)
filt2.Update()

#write to file
writer = itk.ImageFileWriter[itk.Image[pixeltype, dim]].New(filt.GetOutput(), FileName="/home/mates/Desktop/output.jpg")
writer.Update()