Add files via upload

Author: mkang315

mmseg/datasets/MoNuSAC.py

@@ -0,0 +1,33 @@
+import os.path as osp
+
+import mmengine.fileio as fileio
+
+from mmseg.registry import DATASETS
+from .basesegdataset import BaseSegDataset
+
+
+@DATASETS.register_module()
+class MoNuSACDataset(BaseSegDataset):
+    """Pascal VOC dataset.
+
+    Args:
+        split (str): Split txt file for Pascal VOC.
+    """
+    METAINFO = dict(
+        classes=('background', 'Epithelial', 'Lymphocyte', 'Neutrophil','Macrophage'),
+        palette=[[0, 0, 0], [128, 0, 0], [0, 128, 0], [128, 128, 0],
+             [0, 0, 128]])
+       # palette=[[0, 0, 0], [1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]])
+
+    def __init__(self,
+                 ann_file,
+                 img_suffix='.jpg',
+                 seg_map_suffix='.png',
+                 **kwargs) -> None:
+        super().__init__(
+            img_suffix=img_suffix,
+            seg_map_suffix=seg_map_suffix,
+            ann_file=ann_file,
+            **kwargs)
+        assert fileio.exists(self.data_prefix['img_path'],
+                             self.backend_args) and osp.isfile(self.ann_file)

mmseg/datasets/NuInsSeg.py

@@ -0,0 +1,69 @@
+import os.path as osp
+
+import mmengine.fileio as fileio
+
+from mmseg.registry import DATASETS
+from .basesegdataset import BaseSegDataset
+
+
+@DATASETS.register_module()
+class MoNuSACDataset(BaseSegDataset):
+    """Pascal VOC dataset.
+
+    Args:
+        split (str): Split txt file for Pascal VOC.
+    """
+    METAINFO = dict(
+        classes=('background',
+        'human bladder',
+        'human brain',
+        'human cardia',
+        'human cerebellum',
+        'human epiglottis',
+        'human jejunum',
+        'human kidney',
+        'human liver',
+        'human lung',
+        'human melanoma',
+        'human muscle',
+        'human oesophagus',
+        'human pancreas',
+        'human peritoneum',
+        'human placenta',
+        'human pylorus',
+        'human rectum',
+        'human salivory gland',
+        'human spleen',
+        'human testis',
+        'human tongue',
+        'human tonsile',
+        'human umbilical cord',
+        'mouse fat (white and brown)_subscapula',
+        'mouse femur',
+        'mouse heart',
+        'mouse kidney',
+        'mouse liver',
+        'mouse muscle_tibia',
+        'mouse spleen',
+        'mouse thymus' ),
+        palette=[[0, 0, 0],
+                 [128, 0, 0], [0, 128, 0],[0, 0, 128],  [128, 128, 0],[0, 128, 128],  [128, 0, 128], [128, 128, 128],
+                 [255, 0, 0], [0, 255, 0],[0, 0, 255],  [255, 255, 0],[0, 255, 255],  [255, 0, 255], [255, 255, 255],
+                 [64, 0, 0], [0, 64, 0],[0, 0, 64],  [64, 64, 0],[0, 64, 64],  [64, 0, 64], [64, 64, 64],
+                 [192, 0, 0], [0, 192, 0],[0, 0, 192],  [192, 192, 0],[0, 192, 192],  [192, 0, 192], [192, 192, 192],
+                 [32, 0, 0], [0, 32, 0],[0, 0, 32],
+                 ])
+    # palette=[[0, 0, 0], [1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]])
+
+    def __init__(self,
+                 ann_file,
+                 img_suffix='.jpg',
+                 seg_map_suffix='.png',
+                 **kwargs) -> None:
+        super().__init__(
+            img_suffix=img_suffix,
+            seg_map_suffix=seg_map_suffix,
+            ann_file=ann_file,
+            **kwargs)
+        assert fileio.exists(self.data_prefix['img_path'],
+                             self.backend_args) and osp.isfile(self.ann_file)