Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
yaoyuping
nnDetection
Commits
5d93ba19
Commit
5d93ba19
authored
Aug 29, 2021
by
mibaumgartner
Browse files
remove deprecated function
parent
ac9cc5e5
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
23 deletions
+5
-23
nndet/io/itk.py
nndet/io/itk.py
+0
-20
projects/Task021_ProstateX/scripts/prepare.py
projects/Task021_ProstateX/scripts/prepare.py
+5
-3
No files found.
nndet/io/itk.py
View file @
5d93ba19
...
...
@@ -74,26 +74,6 @@ def create_circle_mask_itk(image_itk: sitk.Image,
return
mask_itk
def
copy_meta_data_itk
(
source
:
sitk
.
Image
,
target
:
sitk
.
Image
)
->
sitk
.
Image
:
"""
Copy meta data between files
Args:
source: source file
target: target file
Returns:
sitk.Image: target file with copied meta data
"""
# for i in source.GetMetaDataKeys():
# target.SetMetaData(i, source.GetMetaData(i))
raise
NotImplementedError
(
"Does not work!"
)
target
.
SetOrigin
(
source
.
GetOrigin
())
target
.
SetDirection
(
source
.
GetDirection
())
target
.
SetSpacing
(
source
.
GetSpacing
())
return
target
def
load_sitk
(
path
:
Union
[
Path
,
str
],
**
kwargs
)
->
sitk
.
Image
:
"""
Functional interface to load image with sitk
...
...
projects/Task021_ProstateX/scripts/prepare.py
View file @
5d93ba19
...
...
@@ -11,8 +11,8 @@ from nndet.io.prepare import create_test_split
from
loguru
import
logger
from
nndet.utils.check
import
env_guard
from
nndet.io
import
save_json
,
save_yaml
from
nndet.io.itk
import
load_sitk
,
load_sitk_as_array
,
copy_meta_data_itk
from
nndet.io
import
save_json
from
nndet.io.itk
import
load_sitk
,
load_sitk_as_array
from
nndet.utils.info
import
maybe_verbose_iterable
...
...
@@ -99,7 +99,9 @@ def prepare_case(case_id,
mask
[
_mask
>
0
]
=
idx
mask_final
=
sitk
.
GetImageFromArray
(
mask
)
copy_meta_data_itk
(
t2_data_itk
,
mask_final
)
mask_final
.
SetOrigin
(
t2_data_itk
.
GetOrigin
())
mask_final
.
SetDirection
(
t2_data_itk
.
GetDirection
())
mask_final
.
SetSpacing
(
t2_data_itk
.
GetSpacing
())
df_case
=
df_labels
.
loc
[
df_labels
[
'ProxID'
]
==
case_id
]
instances
=
{}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment