Source code for ddd_pylib.labels_manip._remove_labels_operator

from ._replace_values_operator import ReplaceValuesOperator
from typing import List




class RemoveLabelsOperator(ReplaceValuesOperator):
    """
    Removes a list of labels from a label map by replacing them with 0.
    """
    def __init__(self):
        super().__init__()



    def getPrefix(self):
        return "RemoveLabels-"




    def setRemovedLabels(self, labels: List[int]):
        label_mapping = {label: 0 for label in labels}
        self.setValuesDict(label_mapping)