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")
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")
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
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
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")
def test_get_nlabels_single(self): slab={"liver": 1, "porta": 2} val = ima.get_nlabels(slab, "porta", return_mode="int") self.assertEqual(val, 2)
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)
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")
def test_get_nlabels_single(self): slab = {"liver": 1, "porta": 2} val = ima.get_nlabels(slab, "porta", return_mode="int") self.assertEqual(val, 2)
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)