-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathdk_seg.py
41 lines (24 loc) · 946 Bytes
/
dk_seg.py
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
# -*- coding: utf-8 -*-
"""
"""
from __future__ import division
# import numpy as np
import SimpleITK as sitk
#import os
# from scipy.spatial.distance import directed_hausdorff
# from scipy.spatial.distance import cdist
def resample3d(x, original_spacing, new_spacing, resampling_mode):
'''Resamples the given array into new spacing'''
original_size = x.GetSize()
I_size = [int(spacing/new_s*size)
for spacing, size, new_s in zip(original_spacing, original_size, new_spacing)]
I = sitk.Image(I_size, x.GetPixelIDValue())
I.SetSpacing(new_spacing)
I.SetOrigin(x.GetOrigin())
I.SetDirection(x.GetDirection())
resample = sitk.ResampleImageFilter()
resample.SetReferenceImage(I)
resample.SetInterpolator(resampling_mode)
resample.SetTransform(sitk.Transform())
I = resample.Execute(x)
return I