torchio.transforms.preprocessing.label.label_transform

Complexity

Total Complexity

2

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	LabelTransform.get_images_dict()	0	7	1
A	LabelTransform.get_images()	0	7	1

from typing import List

from ....data.image import LabelMap
from ....data.subject import Subject
from ...transform import Transform


class LabelTransform(Transform):
    """Transform that modifies label maps."""

    def get_images(self, subject: Subject) -> List[LabelMap]:
        images = subject.get_images(
            intensity_only=False,
            include=self.include,
            exclude=self.exclude,
        )
        return [im for im in images if isinstance(im, LabelMap)]

    def get_images_dict(self, subject: Subject) -> List[LabelMap]:
        images = subject.get_images_dict(
            intensity_only=False,
            include=self.include,
            exclude=self.exclude,
        )
        return {k: v for (k, v) in images.items() if isinstance(v, LabelMap)}