Example #1
0
 def test_get_nlabels_multiple(self):
     slab={"liver": 1, "porta": 2}
     val = ima.get_nlabels(slab, [2, "porta", "new", 7], return_mode="str")
     self.assertEqual(val[0], "porta")
     self.assertEqual(val[1], "porta")
     self.assertEqual(val[2], "3")
     self.assertEqual(val[3], "7")
Example #2
0
 def test_get_nlabels_multiple(self):
     slab = {"liver": 1, "porta": 2}
     val = ima.get_nlabels(slab, [2, "porta", "new", 7], return_mode="str")
     self.assertEqual(val[0], "porta")
     self.assertEqual(val[1], "porta")
     self.assertEqual(val[2], "3")
     self.assertEqual(val[3], "7")
Example #3
0
def segmentation_replacement(segmentation,
                             segmentation_new,
                             label,
                             background_label=0,
                             slab=None,
                             label_new=1):
    """
    Remove label from segmentation and put there a new one.

    :param segmentation:
    :param segmentation_new:
    :param label:
    :param background_label:
    :param slab:
    :param label_new:
    :return:
    """
    segmentation_old = ima.select_labels(segmentation, labels=label, slab=slab)
    segmentation[segmentation_old] = ima.get_nlabels(slab, background_label)
    segmentation_new = ima.select_labels(segmentation_new, label_new)
    segmentation[segmentation_new] = ima.get_nlabels(slab, label)
    return segmentation
Example #4
0
def segmentation_replacement(
        segmentation,
        segmentation_new,
        label,
        background_label=0,
        slab=None,
        label_new=1
):
    """
    Remove label from segmentation and put there a new one.

    :param segmentation:
    :param segmentation_new:
    :param label:
    :param background_label:
    :param slab:
    :param label_new:
    :return:
    """
    segmentation_old = ima.select_labels(segmentation, labels=label, slab=slab)
    segmentation[segmentation_old] = ima.get_nlabels(slab, background_label)
    segmentation_new = ima.select_labels(segmentation_new, label_new)
    segmentation[segmentation_new] = ima.get_nlabels(slab, label)
    return segmentation
Example #5
0
    def test_get_nlabels_single_both(self):
        slab={"liver": 1, "porta": 2}

        val = ima.get_nlabels(slab, "porta", return_mode="both")
        self.assertEqual(val[0], 2)
        self.assertEqual(val[1], "porta")
Example #6
0
    def test_get_nlabels_single(self):
        slab={"liver": 1, "porta": 2}

        val = ima.get_nlabels(slab, "porta", return_mode="int")
        self.assertEqual(val, 2)
Example #7
0
 def test_get_nlabels_single_label(self):
     slab={"liver": 1, "kindey": 15, "none":0}
     labels = 1
     val = ima.get_nlabels(slab, labels)
     self.assertEqual(val, 1)
Example #8
0
    def test_get_nlabels_single_both(self):
        slab = {"liver": 1, "porta": 2}

        val = ima.get_nlabels(slab, "porta", return_mode="both")
        self.assertEqual(val[0], 2)
        self.assertEqual(val[1], "porta")
Example #9
0
    def test_get_nlabels_single(self):
        slab = {"liver": 1, "porta": 2}

        val = ima.get_nlabels(slab, "porta", return_mode="int")
        self.assertEqual(val, 2)
Example #10
0
 def test_get_nlabels_single_label(self):
     slab = {"liver": 1, "kindey": 15, "none": 0}
     labels = 1
     val = ima.get_nlabels(slab, labels)
     self.assertEqual(val, 1)