Enable TV MaskRCNN for Geti

src/otx/tools/converter.py

@@ -118,6 +118,10 @@
         "task": OTXTaskType.INSTANCE_SEGMENTATION,
         "model_name": "rtmdet_inst_tiny",
     },
+    "Custom_Instance_Segmentation_MaskRCNN_ResNet50_v2": {
+        "task": OTXTaskType.INSTANCE_SEGMENTATION,
+        "model_name": "maskrcnn_r50_tv",
+    },
     # ROTATED_DETECTION
     "Custom_Rotated_Detection_via_Instance_Segmentation_MaskRCNN_ResNet50": {
         "task": OTXTaskType.ROTATED_DETECTION,

src/otx/tools/templates/detection/instance_segmentation/maskrcnn_r50_tv/template.yaml

@@ -0,0 +1,39 @@
+# Description.
+model_template_id: Custom_Instance_Segmentation_MaskRCNN_ResNet50_v2
+name: Mask R-CNN ResNet-50 v2
+task_type: INSTANCE_SEGMENTATION
+task_family: VISION
+instantiation: "CLASS"
+summary: Instance Segmentation using Mask R-CNN ResNet-50 v2 with an Enhanced Bounding Box Head
+application: ~
+
+# Algo backend.
+framework: OTXDetection v2.9.1
+
+# Capabilities.
+capabilities:
+  - compute_representations
+
+# Hyperparameters.
+hyper_parameters:
+  base_path: ../configuration.yaml
+  parameter_overrides:
+    learning_parameters:
+      batch_size:
+        default_value: 4
+      inference_batch_size:
+        default_value: 1
+      learning_rate:
+        default_value: 0.007
+      num_iters:
+        default_value: 100
+
+# Training resources.
+max_nodes: 1
+training_targets:
+  - GPU
+  - CPU
+
+# Stats.
+gigaflops: 560.5
+size: 177