🐛 Fix HoVer-Net Class Output Information in docs

docs/pretrained.rst

@@ -185,6 +185,15 @@ The input output configuration is as follows:
 
     - hovernet_fast-pannuke
 
+.. collapse:: Output Nuclear Classes
+
+    - 0: Background
+    - 1: Neoplastic
+    - 2: Inflammatory
+    - 3: Connective
+    - 4: Dead
+    - 5: Non-Neoplastic Epithelial
+
 MoNuSAC Dataset
 ---------------
 
@@ -218,6 +227,14 @@ The input output configuration is as follows:
 
     - hovernet_fast-monusac
 
+.. collapse:: Output Nuclear Classes
+
+    - 0: Background
+    - 1: Epithelial
+    - 2: Lymphocyte
+    - 3: Macrophage
+    - 4: Neutrophil
+
 CoNSeP Dataset
 --------------
 
@@ -251,6 +268,14 @@ The input output configuration is as follows:
 
     - hovernet_original-consep
 
+.. collapse:: Output Nuclear Classes
+
+    - 0: Background
+    - 1: Epithelial
+    - 2: Inflammatory
+    - 3: Spindle-Shaped
+    - 4: Miscellaneous
+
 
 .. collapse:: Input Output Configuration Details
 
@@ -275,11 +300,20 @@ The input output configuration is as follows:
 
     - micronet_hovernet-consep
 
+.. collapse:: Output Nuclear Classes
+
+    - 0: Background
+    - 1: Epithelial
+    - 2: Inflammatory
+    - 3: Spindle-Shaped
+    - 4: Miscellaneous
+
 Kumar Dataset
 -------------
 
 We provide the following models trained using the `Kumar dataset <https://monuseg.grand-challenge.org/>`_.
 All model weights trained on Kumar are held under the `Creative Commons Attribution-NonCommercial-ShareAlike Version 4 (CC BY-NC-SA 4.0) License <https://creativecommons.org/licenses/by-nc-sa/4.0/>`_.
+The Kumar dataset does not contain nuclear class information, and so TIAToolbox pretrained models based on Kumar for nuclear segmentation, will only perform segmentation and not classification.
 The input output configuration is as follows:
 
 .. collapse:: Input Output Configuration Details
@@ -428,3 +462,17 @@ The model uses the following input output configuration:
 .. collapse:: Model names
 
     - hovernetplus-oed
+
+.. collapse:: Output Nuclear Classes
+
+    - 0: Background
+    - 1: Other
+    - 2: Epithelial
+
+.. collapse:: Output Region Classes
+
+    - 0: Background
+    - 1: Other Tissue
+    - 2: Basal Epithelium
+    - 3: (Core) Epithelium
+    - 4: Keratin

tiatoolbox/data/pretrained_model.yaml

@@ -643,6 +643,13 @@ hovernet_fast-pannuke:
         kwargs:
             num_types: 6
             mode: "fast"
+            nuc_type_dict: {
+                0: "Background",
+                1: "Epithelial",
+                2: "Inflammatory",
+                3: "Spindle-Shaped",
+                4: "Miscellaneous",
+            }
     ioconfig:
         class: semantic_segmentor.IOSegmentorConfig
         kwargs:
@@ -666,6 +673,13 @@ hovernet_fast-monusac:
         kwargs:
             num_types: 5
             mode: "fast"
+            nuc_type_dict: {
+                0: "Background",
+                1: "Epithelial",
+                2: "Lymphocyte",
+                3: "Macrophage",
+                4: "Neutrophil",
+            }
     ioconfig:
         class: semantic_segmentor.IOSegmentorConfig
         kwargs:
@@ -689,6 +703,13 @@ hovernet_original-consep:
         kwargs:
             num_types: 5
             mode: "original"
+            nuc_type_dict: {
+                0: "Background",
+                1: "Epithelial",
+                2: "Inflammatory",
+                3: "Spindle-Shaped",
+                4: "Miscellaneous",
+            }
     ioconfig:
         class: semantic_segmentor.IOSegmentorConfig
         kwargs:
@@ -734,6 +755,18 @@ hovernetplus-oed:
         kwargs:
             num_types: 3
             num_layers: 5
+            nuc_type_dict: {
+                0: "Background",
+                1: "Other",
+                2: "Epithelial",
+            }
+            layer_type_dict: {
+                0: "Background",
+                1: "Other Tissue",
+                2: "Basal Epithelium",
+                3: "(Core) Epithelium",
+                4: "Keratin",
+            }
     ioconfig:
         class: semantic_segmentor.IOSegmentorConfig
         kwargs: