Skip to content

SaveImage set_options to return self #6146

Closed
@wyli

Description

@wyli

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 request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions