-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Closed
Labels
enhancementNew feature or requestNew feature or request
Description
Describe the bug
follow up of #6136, SaveImage.set_options is not easy to use in a compose transform:
saver = SaveImaged(keys="pred")
saver.set_options(init_kwargs={"affine_lps_to_ras": True})
postprocessing = Compose(
[
EnsureTyped(keys=["pred", "seg"]), # ensure Tensor type after `decollate`
Activationsd(keys="pred", sigmoid=True),
...
AsDiscreted(keys="pred", threshold=0.5),
saver,
]
)this is a feature request to support:
postprocessing = Compose(
[
EnsureTyped(keys=["pred", "seg"]), # ensure Tensor type after `decollate`
Activationsd(keys="pred", sigmoid=True),
...
AsDiscreted(keys="pred", threshold=0.5),
SaveImaged(keys="pred").set_options(init_kwargs={"affine_lps_to_ras": True}),
]
)which is easily achievable if SaveImage.set_options can return self
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request