{
  "architecture": "fast_base",
  "architecture_id": "ocr.fast",
  "backbone": {
    "feature_channels": [
      64,
      128,
      256,
      512
    ],
    "feature_layers": [
      "3",
      "4",
      "5",
      "6"
    ],
    "name": "textnet_base",
    "stages": [
      {
        "in_channels": [
          64,
          64,
          64,
          64,
          64,
          64,
          64,
          64,
          64,
          64
        ],
        "kernel_size": [
          [
            3,
            3
          ],
          [
            3,
            3
          ],
          [
            3,
            1
          ],
          [
            3,
            3
          ],
          [
            3,
            1
          ],
          [
            3,
            3
          ],
          [
            3,
            3
          ],
          [
            1,
            3
          ],
          [
            3,
            3
          ],
          [
            3,
            3
          ]
        ],
        "out_channels": [
          64,
          64,
          64,
          64,
          64,
          64,
          64,
          64,
          64,
          64
        ],
        "stride": [
          1,
          2,
          1,
          1,
          1,
          1,
          1,
          1,
          1,
          1
        ]
      },
      {
        "in_channels": [
          64,
          128,
          128,
          128,
          128,
          128,
          128,
          128,
          128,
          128
        ],
        "kernel_size": [
          [
            3,
            3
          ],
          [
            1,
            3
          ],
          [
            3,
            3
          ],
          [
            3,
            1
          ],
          [
            3,
            3
          ],
          [
            3,
            3
          ],
          [
            3,
            1
          ],
          [
            3,
            1
          ],
          [
            3,
            3
          ],
          [
            3,
            3
          ]
        ],
        "out_channels": [
          128,
          128,
          128,
          128,
          128,
          128,
          128,
          128,
          128,
          128
        ],
        "stride": [
          2,
          1,
          1,
          1,
          1,
          1,
          1,
          1,
          1,
          1
        ]
      },
      {
        "in_channels": [
          128,
          256,
          256,
          256,
          256,
          256,
          256,
          256
        ],
        "kernel_size": [
          [
            3,
            3
          ],
          [
            3,
            3
          ],
          [
            3,
            3
          ],
          [
            1,
            3
          ],
          [
            3,
            3
          ],
          [
            3,
            1
          ],
          [
            3,
            3
          ],
          [
            3,
            1
          ]
        ],
        "out_channels": [
          256,
          256,
          256,
          256,
          256,
          256,
          256,
          256
        ],
        "stride": [
          2,
          1,
          1,
          1,
          1,
          1,
          1,
          1
        ]
      },
      {
        "in_channels": [
          256,
          512,
          512,
          512,
          512
        ],
        "kernel_size": [
          [
            3,
            3
          ],
          [
            1,
            3
          ],
          [
            3,
            1
          ],
          [
            3,
            1
          ],
          [
            1,
            3
          ]
        ],
        "out_channels": [
          512,
          512,
          512,
          512,
          512
        ],
        "stride": [
          2,
          1,
          1,
          1,
          1
        ]
      }
    ]
  },
  "checkpoint_tensors": {
    "ignored_patterns": [
      "*.num_batches_tracked"
    ],
    "ignored_scalar_or_non_tensor": 131,
    "saved": 626
  },
  "forward_contract": {
    "exportable": true,
    "input": {
      "dtype": "f32",
      "name": "input",
      "shape": [
        1,
        3,
        1024,
        1024
      ]
    },
    "output": {
      "dtype": "f32",
      "name": "logits",
      "shape": [
        1,
        1,
        1024,
        1024
      ]
    },
    "postprocess_probability_map": "sigmoid(logits)"
  },
  "head": {
    "dropout": 0.1,
    "final_conv_bias": false,
    "hidden_channels": 128,
    "input_channels": 512,
    "name": "FastHead",
    "num_classes": 1
  },
  "markers_postprocessor": {
    "bin_thresh": 0.1,
    "box_thresh": 0.1,
    "decode": "decode_prob_map",
    "note": "Existing markers FAST decode keeps the DBNet unclip ratio pending parity measurement.",
    "unclip_ratio": 1.5
  },
  "model_type": "doctr-fast",
  "neck": {
    "input_channel_scales": [
      1,
      2,
      4,
      8
    ],
    "name": "FastNeck",
    "out_channels": 128,
    "upsample": "bilinear_to_first_feature"
  },
  "source": {
    "checkpoint_filename": "fast_base-688a8b34.pt",
    "checkpoint_sha256": "688a8b3489e9f5d0290c476c6272ec3b18de3ee646c8a0dc158203b1a9c62ace",
    "checkpoint_size_bytes": 65814772,
    "checkpoint_url": "https://doctr-static.mindee.com/models?id=v0.8.1/fast_base-688a8b34.pt&src=0",
    "repo": "mindee/doctr",
    "revision": "6541a6e1e39259b3caea5e0ee94345ff4395f25b",
    "tag": "v1.0.1"
  },
  "upstream_postprocessor": {
    "assume_straight_pages": true,
    "bin_thresh": 0.1,
    "box_thresh": 0.1,
    "unclip_ratio": 1.0
  }
}