torchio.transforms.preprocessing.label.remove_labels.RemoveLabels.__init__() - Code Metrics - Inspection of "Add label remapping transforms (#402)" - fepegar/torchio - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( b9ac52...6aebda )

by Fernando

created 2020-12-29 14:50 UTC

RemoveLabels.init() A

↳ Parent: torchio.transforms.preprocessing.label.remove_labels

Complexity

Conditions

Size

Total Lines	13
Code Lines	12

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	1
eloc	12
nop	5
dl	0
loc	13
rs	9.8
c	0
b	0
f	0

from typing import Sequence

from ...transform import TypeMaskingMethod
from .remap_labels import RemapLabels


class RemoveLabels(RemapLabels):
    r"""Remove labels from a label map by remapping them to the background label.

    This transformation is not `invertible <invertibility>`_.

    Args:
        labels: A sequence of label integers that will be removed.
        background_label: integer that specifies which label is considered to be
            background (generally 0).
        masking_method: See :class:`~torchio.RemapLabels`.
        **kwargs: See :class:`~torchio.transforms.Transform` for additional
            keyword arguments.
    """
    def __init__(
            self,
            labels: Sequence[int],
            background_label: int = 0,
            masking_method: TypeMaskingMethod = None,
            **kwargs
            ):
        remapping = {label: background_label for label in labels}
        super().__init__(remapping, masking_method, **kwargs)
        self.labels = labels
        self.background_label = background_label
        self.masking_method = masking_method
        self.args_names = ('labels', 'background_label', 'masking_method',)

    def is_invertible(self):
        return False


1			from typing import Sequence
2
3			from ...transform import TypeMaskingMethod
4			from .remap_labels import RemapLabels
5
6
7			class RemoveLabels(RemapLabels):
8			r"""Remove labels from a label map by remapping them to the background label.
9
10			This transformation is not `invertible <invertibility>`_.
11
12			Args:
13			labels: A sequence of label integers that will be removed.
14			background_label: integer that specifies which label is considered to be
15			background (generally 0).
16			masking_method: See :class:`~torchio.RemapLabels`.
17			**kwargs: See :class:`~torchio.transforms.Transform` for additional
18			keyword arguments.
19			"""
20			def __init__(
21			self,
22			labels: Sequence[int],
23			background_label: int = 0,
24			masking_method: TypeMaskingMethod = None,
25			**kwargs
26			):
27			remapping = {label: background_label for label in labels}
28			super().__init__(remapping, masking_method, **kwargs)
29			self.labels = labels
30			self.background_label = background_label
31			self.masking_method = masking_method
32			self.args_names = ('labels', 'background_label', 'masking_method',)
33
34			def is_invertible(self):
35			return False
36

fepegar / torchio

Push — master ( b9ac52...6aebda )

RemoveLabels.__init__() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like

RemoveLabels.init() A