bundle.json

{
  "architecture_id": "ocr.fast",
  "bundle_id": "ocr/detector/fast",
  "license": "Apache-2.0",
  "parity_source": {
    "files": [
      "model.onnx",
      "config.json"
    ],
    "repo": "Felix92/onnxtr-fast-base",
    "revision": "9e1b4f4773acfee2f252523c7fa563bf699c1fe5"
  },
  "provenance": {
    "checkpoint_url": "https://doctr-static.mindee.com/models?id=v0.8.1/fast_base-688a8b34.pt&src=0",
    "conversion": "PyTorch state_dict tensors copied to safetensors without transpose",
    "runtime_onnx": "false",
    "source_tag": "v1.0.1"
  },
  "required_files": [
    {
      "path": "model.safetensors",
      "role": "weights"
    },
    {
      "path": "config.json",
      "role": "config"
    },
    {
      "path": "preprocessor_config.json",
      "role": "preprocessor"
    },
    {
      "path": "tensor_map.json",
      "role": "tensor-map"
    },
    {
      "path": "validation.json",
      "role": "validation"
    }
  ],
  "schema_version": 1,
  "source": {
    "files": [
      "doctr/models/detection/fast/pytorch.py",
      "doctr/models/classification/textnet/pytorch.py",
      "doctr/models/modules/layers/pytorch.py",
      "fast_base-688a8b34.pt"
    ],
    "repo": "mindee/doctr",
    "revision": "6541a6e1e39259b3caea5e0ee94345ff4395f25b"
  },
  "tensor_contract": {
    "inputs": [
      {
        "dtype": "f32",
        "name": "input",
        "shape": [
          1,
          3,
          1024,
          1024
        ]
      }
    ],
    "outputs": [
      {
        "dtype": "f32",
        "name": "logits",
        "shape": [
          1,
          1,
          1024,
          1024
        ]
      }
    ]
  },
  "validated_runtime": [
    {
      "device": "cpu",
      "dtype": "f32"
    }
  ]
}