{
  "model": "modnet",
  "upstream": {
    "model": "ZHKKKe/MODNet (photographic portrait matting checkpoint)",
    "weights": {
      "repo": "boopathiraj/MODNet",
      "revision": "a967175f8fd63de67a1c161d8daac966d2b57e3a",
      "file": "modnet/pretrained/modnet_photographic_portrait_matting.ckpt",
      "sha256": "7c22235f0925deba15d4d63e53afcb654c47055bbcd98f56e393ab2584007ed8",
      "official": false,
      "note": "community mirror; official checkpoints are on Google Drive, linked from GitHub, neither reachable from the export host"
    },
    "architecture_source": {
      "repo": "boopathiraj/MODNet",
      "revision": "a967175f8fd63de67a1c161d8daac966d2b57e3a",
      "files_sha256": {
        "modnet/__init__.py": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
        "modnet/modnet.py": "8fba11b0fe1c5e10ed5feb8c23e48543c2bf32242eda32621084df3624183109",
        "modnet/src/__init__.py": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
        "modnet/src/models/__init__.py": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
        "modnet/src/models/backbones/__init__.py": "28a5fb95f7dcf9e365edbf42c6d2e8ea0ca4839e51fd7f11bd0547d2359fcd96",
        "modnet/src/models/backbones/wrapper.py": "41197be7eb96b8a60dc034b55d8c9340dd682a41441dcf2ce67238955dfa5607",
        "modnet/src/models/backbones/mobilenetv2.py": "e3cc8ad6a9933ba18a17a62d5f887c64e0721240871ea8b48742fb9a8a2c3199",
        "modnet/LICENSE": "c71d239df91726fc519c6eb72d318ec65820627232b2f796219e87dcf35d0ab4"
      },
      "official": false,
      "note": "copy of the upstream src/models tree with modnet.py moved to the package root"
    },
    "license": "Apache-2.0"
  },
  "versions": {
    "python": "3.12.15+meta",
    "torch": "2.14.1+cpu",
    "torchao": "0.18.0+cpu",
    "executorch": "1.5.1+cpu",
    "executorch_git": "3b60683923245cf472b7323426920e15623ba361",
    "transformers": "5.19.0",
    "huggingface_hub": "2.2.0"
  },
  "export": {
    "api": "torch.export.export(strict=False) -> to_edge_transform_and_lower(partitioner=[WebGPUPartitioner()]) -> to_executorch()",
    "partitioner": "executorch.backends.webgpu.partitioner.WebGPUPartitioner() (default compile options; wraps VulkanPartitioner)",
    "inputs": [
      {
        "shape": [
          1,
          3,
          256,
          256
        ],
        "dtype": "float32",
        "layout": "NCHW"
      }
    ],
    "outputs": [
      {
        "shape": [
          1,
          1,
          256,
          256
        ],
        "dtype": "float32"
      }
    ],
    "dtype": "float32",
    "quantization": "none (fp32)",
    "variant": "photographic",
    "preprocessing_outside_graph": "RGB / 255, then (x - 0.5) / 0.5",
    "postprocessing_outside_graph": "alpha in [0, 1]"
  },
  "files": {
    "modnet_webgpu.pte": {
      "bytes": 26109828,
      "sha256": "0c2ff92e26087e98b4b2b61497d48d278be2474cc3cfabbc16f58da7eb5ad6c4"
    },
    "modnet_portable.pte": {
      "bytes": 26193908,
      "sha256": "e74de105325321efeb5ef52d27a27c7acae2829d2098572a0be6c6dea7ac7dbd",
      "note": "CPU-only validation artifact (portable kernels, no delegate); not for release"
    }
  },
  "graph": {
    "non_delegated_kernels": {},
    "delegates": [
      {
        "id": "VulkanBackend",
        "ops": {
          "et_vk.prepack.default": 16,
          "et_vk.conv_with_clamp.default": 35,
          "aten.convolution.default": 34,
          "aten.add.Tensor": 26,
          "aten.mean.dim": 33,
          "aten.clone.default": 2,
          "aten.view_copy.default": 2,
          "aten.linear.default": 2,
          "aten.relu.default": 17,
          "aten.sigmoid.default": 2,
          "aten.expand_copy.default": 1,
          "aten.mul.Tensor": 17,
          "aten.upsample_bilinear2d.vec": 9,
          "aten.slice_copy.Tensor": 32,
          "aten._native_batch_norm_legit_no_training.default": 16,
          "aten.sub.Tensor": 16,
          "aten.sqrt.default": 16,
          "aten.div.Tensor": 16,
          "aten.cat.default": 22
        },
        "inputs": 1,
        "outputs": 1
      }
    ]
  },
  "command": "python recipes/modnet.py --out out/modnet"
}