| { |
| "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 |
| } |
| } |
|
|
