Add zip models

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+sam2_hiera_base_plus.zip filter=lfs diff=lfs merge=lfs -text
+sam2_hiera_large.zip filter=lfs diff=lfs merge=lfs -text
+sam2_hiera_small.zip filter=lfs diff=lfs merge=lfs -text
+sam2_hiera_tiny.zip filter=lfs diff=lfs merge=lfs -text

sam2_hiera_tiny_decoder.onnx → sam2_hiera_base_plus.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e6503974464a1e9766f1978f8f14b36369d1fe939a616ebeb7bad9278c1e57d0
-size 16531251
+oid sha256:c56282103c2bf99bdab07d2dddb1be67c03d71659ade3f9dcf1d7001fda582fb
+size 360422777

sam2_hiera_tiny_encoder.onnx → sam2_hiera_large.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dfc15530ff902a9d8f97ea04b262aad57feea9aa3a5895b3061dd268be964dfb
-size 109471936
+oid sha256:a967ef6e54794e9494c8b71201766532b7b2929fe18f56b8e63940e97dc671ac
+size 910003530

sam2_hiera_small.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ef9047eb7fc7e36041c88a9f968c3c0b1d1640b88ef66d754b7d558a10cee38
+size 183345019

sam2_hiera_tiny.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7454c3afd835b2acaad863afe3acb11f4e4af039c96e886989ad3873a338e1ec
+size 154902833

zip_models.py

@@ -0,0 +1,38 @@
+import os
+import zipfile
+import yaml
+
+version = "20240803"
+base_name_to_model_name = {
+    "sam2_hiera_tiny": "Segment Anything 2 (Hiera-Tiny)",
+    "sam2_hiera_small": "Segment Anything 2 (Hiera-Small)",
+    "sam2_hiera_base_plus": "Segment Anything 2 (Hiera-Base+)",
+    "sam2_hiera_large": "Segment Anything 2 (Hiera-Large)",
+}
+
+model_list = {}
+onnx_files = [f for f in os.listdir(".") if f.endswith(".onnx")]
+for onnx_file in onnx_files:
+    model_base_name, model_part, _ = onnx_file.split(".")
+    if model_base_name not in model_list:
+        model_list[model_base_name] = {}
+        model_list[model_base_name]["type"] = "segment_anything"
+        model_list[model_base_name]["input_size"] = 1024
+        model_list[model_base_name]["max_width"] = 1024
+        model_list[model_base_name]["max_height"] = 1024
+    if model_part == "encoder":
+        model_list[model_base_name]["encoder_model_path"] = onnx_file
+    elif model_part == "decoder":
+        model_list[model_base_name]["decoder_model_path"] = onnx_file
+    model_list[model_base_name]["basename"] = model_base_name
+    model_list[model_base_name]["name"] = model_base_name + "_" + version
+    model_list[model_base_name]["display_name"] = base_name_to_model_name[model_base_name]
+
+for model in model_list.values():
+    output_zip = model.pop("basename") + ".zip"
+    with zipfile.ZipFile(output_zip, "w") as z:
+        z.write(model["encoder_model_path"])
+        z.write(model["decoder_model_path"])
+        # Save config in yaml
+        with z.open("config.yaml", "w") as f:
+            f.write(yaml.dump(model).encode("utf-8"))