🔨 [Add] wegith transform for v9seg model

yolo/tools/format_converters.py

@@ -83,3 +83,55 @@ def convert_weight_v7(old_state_dict, new_state_dict):
         assert new_shape == old_shape, "Weight Shape Mismatch!! {old_key_name}"
         new_state_dict[new_key_name] = old_state_dict[old_key_name]
     return new_state_dict
+
+
+replace_dict = {"cv": "conv", ".m.": ".bottleneck."}
+
+
+def convert_weight_seg(old_state_dict, new_state_dict):
+    diff = -1
+    for old_weight_name in old_state_dict.keys():
+        old_idx = int(old_weight_name.split(".")[1])
+        if old_idx == 23:
+            diff = 3
+        elif old_idx == 41:
+            diff = -19
+        new_idx = old_idx + diff
+        new_weight_name = old_weight_name.replace(f".{old_idx}.", f".{new_idx}.")
+        for key, val in replace_dict.items():
+            new_weight_name = new_weight_name.replace(key, val)
+
+        if new_weight_name not in new_state_dict.keys():
+            heads = "heads"
+            _, _, conv_name, conv_idx, *details = old_weight_name.split(".")
+            if "proto" in conv_name:
+                conv_idx = "3"
+                new_weight_name = ".".join(["model", str(layer_idx), heads, conv_task, *details])
+                continue
+            if "dfl" in old_weight_name:
+                continue
+            if conv_name == "cv2" or conv_name == "cv3" or conv_name == "cv6":
+                layer_idx = 44
+                heads = "detect.heads"
+            if conv_name == "cv4" or conv_name == "cv5" or conv_name == "cv7":
+                layer_idx = 25
+                heads = "detect.heads"
+
+            if conv_name == "cv2" or conv_name == "cv4":
+                conv_task = "anchor_conv"
+            if conv_name == "cv3" or conv_name == "cv5":
+                conv_task = "class_conv"
+            if conv_name == "cv6" or conv_name == "cv7":
+                conv_task = "mask_conv"
+                heads = "heads"
+
+            new_weight_name = ".".join(["model", str(layer_idx), heads, conv_idx, conv_task, *details])
+
+        if (
+            new_weight_name not in new_state_dict.keys()
+            or new_state_dict[new_weight_name].shape != old_state_dict[old_weight_name].shape
+        ):
+            print(f"new: {new_weight_name}, old: {old_weight_name}")
+            print(f"{new_state_dict[new_weight_name].shape} {old_state_dict[old_weight_name].shape}")
+        new_state_dict[new_weight_name] = old_state_dict[old_weight_name]
+    return new_state_dict