MCPcopy Create free account

hub / github.com/mindee/doctr / functions

Functions871 in github.com/mindee/doctr

⨍ Functions 871◇ Types & classes 153↳ Endpoints 52
Method__init__
(self, transforms: list[Callable[[Any, Any], tuple[Any, Any]]])
doctr/transforms/modules/base.py:37
Method__init__
(self, transform: Callable[[Any], Any])
doctr/transforms/modules/base.py:63
Method__init__
(self, min_val: float = 0.5)
doctr/transforms/modules/base.py:86
Method__init__
(self, transforms: list[Callable[[Any], Any]])
doctr/transforms/modules/base.py:112
Method__init__
(self, transform: Callable[[Any], Any], p: float = 0.5)
doctr/transforms/modules/base.py:137
Method__init__
(self, max_angle: float = 5.0, expand: bool = False)
doctr/transforms/modules/base.py:161
Method__init__
(self, scale: tuple[float, float] = (0.08, 1.0), ratio: tuple[float, float] = (0.75, 1.33))
doctr/transforms/modules/base.py:184
Method__init__
( self, img_folder: str, label_path: str, train: bool = True, **kwargs
doctr/datasets/mjsynth.py:75
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/svt.py:41
Method__init__
( self, train: bool = True, use_polygons: bool = False, **kwargs: Any, )
doctr/datasets/doc_artefacts.py:38
Method__init__
( self, img_folder: str, label_path: str, train: bool = True, **kwargs
doctr/datasets/iiithws.py:41
Method__init__
( self, img_folder: str, label_path: str, use_polygons: bool = False,
doctr/datasets/detection.py:35
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/svhn.py:50
Method__init__
( self, img_folder: str, label_folder: str, use_polygons: bool = False,
doctr/datasets/ic13.py:44
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/cord.py:51
Method__init__
( self, img_folder: str, labels_path: str, **kwargs: Any, )
doctr/datasets/recognition.py:30
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/funsd.py:43
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/ic03.py:49
Method__init__
( self, img_folder: str, label_path: str, train: bool = True, use_poly
doctr/datasets/imgur5k.py:52
Method__init__
( self, img_folder: str, label_file: str, use_polygons: bool = False,
doctr/datasets/ocr.py:33
Method__init__
( self, img_folder: str, label_path: str, train: bool = True, use_poly
doctr/datasets/coco_text.py:45
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/synthtext.py:50
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/iiit5k.py:44
Method__init__
( self, img_folder: str, **kwargs: Any, )
doctr/datasets/orientation.py:28
Method__init__
( self, img_folder: str, label_path: str, train: bool = True, use_poly
doctr/datasets/wildreceipt.py:48
Method__init__
( self, train: bool = True, use_polygons: bool = False, recognition_task: bool
doctr/datasets/sroie.py:50
Method__init__
(self, *args, **kwargs)
doctr/datasets/generator/pytorch.py:29
Method__init__
( self, vocab: str, num_samples: int, cache_samples: bool = False, fon
doctr/datasets/generator/base.py:58
Method__init__
( self, vocab: str, min_chars: int, max_chars: int, num_samples: int,
doctr/datasets/generator/base.py:105
Method__init__
( self, root: str | Path, img_transforms: Callable[[Any], Any] | None = None,
doctr/datasets/datasets/base.py:26
Method__init__
( self, url: str, file_name: str | None = None, file_hash: str | None = None,
doctr/datasets/datasets/base.py:96
Method__init__
(self, cfg: dict[str, Any] | None = None)
doctr/models/core.py:17
Method__init__
( self, resolve_lines: bool = True, resolve_blocks: bool = False, paragraph_br
doctr/models/builder.py:30
Method__init__
(self, output_size)
doctr/models/modules/layers/pytorch.py:42
Method__init__
( self, in_channels: int, out_channels: int, kernel_size: int | tuple[int, int
doctr/models/modules/layers/pytorch.py:65
Method__init__
(self, d_model: int, dropout: float = 0.1, max_len: int = 5000)
doctr/models/modules/transformer/pytorch.py:21
Method__init__
( self, d_model: int, ffd: int, dropout: float = 0.1, activation_fct: Callable[[Any], Any] = nn.ReLU()
doctr/models/modules/transformer/pytorch.py:61
Method__init__
(self, num_heads: int, d_model: int, dropout: float = 0.1)
doctr/models/modules/transformer/pytorch.py:76
Method__init__
( self, num_layers: int, num_heads: int, d_model: int, dff: int, # hi
doctr/models/modules/transformer/pytorch.py:107
Method__init__
(self, input_shape: tuple[int, int, int], embed_dim: int, patch_size: tuple[int, int])
doctr/models/modules/vision_transformer/pytorch.py:17
Method__init__
( self, vocab: str, )
doctr/models/recognition/core.py:46
Method__init__
(self, in_feats: int, rnn_units: int, dropout_prob: float = 0.0)
doctr/models/recognition/sar/pytorch.py:35
Method__init__
(self, feat_chans: int, state_chans: int, attention_units: int)
doctr/models/recognition/sar/pytorch.py:48
Method__init__
( self, rnn_units: int, max_length: int, vocab_size: int, embedding_un
doctr/models/recognition/sar/pytorch.py:92
Method__init__
(self, vocab_size: int, d_model: int)
doctr/models/recognition/parseq/pytorch.py:46
Method__init__
( self, d_model: int, num_heads: int = 12, ffd: int = 2048, ffd_ratio:
doctr/models/recognition/parseq/pytorch.py:66
Method__init__
( self, vocab: str, )
doctr/models/recognition/parseq/base.py:49
Method__init__
( self, pre_processor: PreProcessor, model: nn.Module, split_wide_crops: bool
doctr/models/recognition/predictor/pytorch.py:30
Method__init__
( self, feature_extractor: nn.Module, vocab: str, rnn_units: int = 128,
doctr/models/recognition/crnn/pytorch.py:112
Method__init__
( self, feature_extractor: nn.Module, vocab: str, d_model: int = 512,
doctr/models/recognition/master/pytorch.py:54
Method__init__
( self, vocab: str, )
doctr/models/recognition/master/base.py:49
Method__init__
( self, feature_extractor: nn.Module, vocab: str, input_shape: tuple[int, int,
doctr/models/recognition/viptr/pytorch.py:98
Method__init__
( self, feature_extractor, vocab: str, embedding_units: int, max_lengt
doctr/models/recognition/vitstr/pytorch.py:56
Method__init__
( self, vocab: str, )
doctr/models/recognition/vitstr/base.py:48
Method__init__
(self, box_thresh: float = 0.5, bin_thresh: float = 0.5, assume_straight_pages: bool = True)
doctr/models/detection/core.py:24
Method__init__
(self, layer_shapes: list[tuple[int, int, int]])
doctr/models/detection/linknet/pytorch.py:47
Method__init__
( self, bin_thresh: float = 0.1, box_thresh: float = 0.1, assume_straight_page
doctr/models/detection/linknet/base.py:30
Method__init__
( self, pre_processor: PreProcessor, model: nn.Module, )
doctr/models/detection/predictor/pytorch.py:27
Method__init__
( self, in_channels: int, out_channels: int = 128, )
doctr/models/detection/fast/pytorch.py:55
Method__init__
( self, in_channels: int, num_classes: int, out_channels: int = 128, d
doctr/models/detection/fast/pytorch.py:86
Method__init__
( self, bin_thresh: float = 0.1, box_thresh: float = 0.1, assume_straight_page
doctr/models/detection/fast/base.py:30
Method__init__
( self, in_channels: list[int], out_channels: int, deform_conv: bool = False,
doctr/models/detection/differentiable_binarization/pytorch.py:49
Method__init__
( self, box_thresh: float = 0.1, bin_thresh: float = 0.3, assume_straight_page
doctr/models/detection/differentiable_binarization/base.py:32
Method__init__
( self, det_predictor: DetectionPredictor, reco_predictor: RecognitionPredictor,
doctr/models/predictor/pytorch.py:41
Method__init__
( self, assume_straight_pages: bool = True, straighten_pages: bool = False, pr
doctr/models/predictor/base.py:40
Method__init__
( self, d_model: int, num_layers: int, num_heads: int, ffd_ratio: int,
doctr/models/classification/vit/pytorch.py:78
Method__init__
( self, num_blocks: list[int], output_channels: list[int], stage_stride: list[
doctr/models/classification/resnet/pytorch.py:100
Method__init__
( self, inplanes: int, headers: int = 8, attn_scale: bool = False, rat
doctr/models/classification/magc_resnet/pytorch.py:45
Method__init__
( self, pre_processor: PreProcessor | None, model: nn.Module | None, )
doctr/models/classification/predictor/pytorch.py:26
Method__init__
( self, stages: list[dict[str, list[int]]], input_shape: tuple[int, int, int] = (3, 32
doctr/models/classification/textnet/pytorch.py:56
Method__init__
(self, in_features: int, out_features: int)
doctr/models/classification/vip/pytorch.py:49
Method__init__
Args: embed_dim: dimension of embeddings local_unit: local mixing block(s) global_unit: global mixing blo
doctr/models/classification/vip/pytorch.py:60
Method__init__
(self, dims: tuple[int, int, int, int] = (0, 2, 3, 1))
doctr/models/classification/vip/layers/pytorch.py:30
Method__init__
(self, dim: int = 3)
doctr/models/classification/vip/layers/pytorch.py:41
Method__init__
(self, in_channels: int = 3, embed_dim: int = 128)
doctr/models/classification/vip/layers/pytorch.py:61
Method__init__
( self, dim: int, num_heads: int, mlp_ratio: float = 4.0, qkv_bias: bo
doctr/models/classification/vip/layers/pytorch.py:160
Method__init__
( self, dim: int, num_heads: int = 1, qk_scale: float | None = None, a
doctr/models/classification/vip/layers/pytorch.py:213
Method__init__
( self, dim: int = 64, sr_ratio: int = 1, num_heads: int = 1, mlp_rati
doctr/models/classification/vip/layers/pytorch.py:298
Method__init__
(self, dim: int, out_dim: int)
doctr/models/classification/vip/layers/pytorch.py:344
Method__init__
( self, dim: int, idx: int, split_size: int = 7, dim_out: int | None =
doctr/models/classification/vip/layers/pytorch.py:384
Method__init__
( self, dim: int, num_heads: int, split_size: int = 7, mlp_ratio: floa
doctr/models/classification/vip/layers/pytorch.py:562
Method__init__
( self, det_predictor: DetectionPredictor, reco_predictor: RecognitionPredictor,
doctr/models/kie_predictor/pytorch.py:41
Method__init__
( self, assume_straight_pages: bool = True, straighten_pages: bool = False, pr
doctr/models/kie_predictor/base.py:35
Method__init__
( self, output_size: tuple[int, int], batch_size: int, mean: tuple[float, floa
doctr/models/preprocessor/pytorch.py:32
Method__len__
(self)
doctr/datasets/generator/base.py:88
Method__len__
(self)
doctr/datasets/generator/base.py:142
Method__len__
(self)
doctr/datasets/datasets/base.py:42
Method__repr__
(self)
doctr/transforms/modules/pytorch.py:118
Method__repr__
(self)
doctr/datasets/datasets/base.py:78
Method_init_weights
(self, m)
doctr/models/classification/vip/pytorch.py:222
Method_postprocess
(decoded_features: torch.Tensor)
doctr/models/recognition/sar/pytorch.py:276
Method_postprocess
(logits: torch.Tensor)
doctr/models/recognition/parseq/pytorch.py:395
Method_postprocess
(logits: torch.Tensor)
doctr/models/recognition/crnn/pytorch.py:227
Method_postprocess
(logits: torch.Tensor)
doctr/models/recognition/master/pytorch.py:223
Method_postprocess
(decoded_features: torch.Tensor)
doctr/models/recognition/viptr/pytorch.py:170
Method_postprocess
(decoded_features: torch.Tensor)
doctr/models/recognition/vitstr/pytorch.py:121
Method_postprocess
(prob_map: torch.Tensor)
doctr/models/detection/linknet/pytorch.py:197
Method_postprocess
(prob_map: torch.Tensor)
doctr/models/detection/fast/pytorch.py:210
Method_postprocess
(prob_map: torch.Tensor)
doctr/models/detection/differentiable_binarization/pytorch.py:219
Method_read_sample
(self, index: int)
doctr/datasets/generator/base.py:91
← previousnext →401–500 of 871, ranked by callers