deepdoctection 1.0.2__tar.gz → 1.0.4__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/PKG-INFO +1 -1
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/pyproject.toml +1 -1
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/__init__.py +1 -1
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/d2detect.py +5 -2
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/PKG-INFO +1 -1
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/README.md +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/setup.cfg +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/__init__.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/config.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/dd.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/factory.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/__init__.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/conf_dd_one.yaml +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/conf_tesseract.yaml +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/profiles.jsonl +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/__init__.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/accmetric.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/base.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/cocometric.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/eval.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/registry.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/tedsmetric.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/__init__.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/base.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/deskew.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/doctrocr.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/hfdetr.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/hflayoutlm.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/hflm.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/model.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/pdftext.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/tessocr.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/texocr.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/__init__.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/anngen.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/base.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/common.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/concurrency.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/doctectionpipe.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/language.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/layout.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/lm.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/order.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/refine.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/registry.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/segment.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/sub_layout.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/text.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/transform.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/py.typed +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/__init__.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/d2_frcnn_train.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/hf_detr_train.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/hf_layoutlm_train.py +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/SOURCES.txt +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/dependency_links.txt +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/requires.txt +0 -0
- {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/top_level.txt +0 -0
|
@@ -12,7 +12,7 @@ from dd_core.utils.env_info import collect_env_info
|
|
|
12
12
|
from dd_core.utils.file_utils import _LazyModule
|
|
13
13
|
from dd_core.utils.logger import LoggingRecord, logger
|
|
14
14
|
|
|
15
|
-
__version__ = "1.0.
|
|
15
|
+
__version__ = "1.0.4"
|
|
16
16
|
_IMPORT_STRUCTURE = {
|
|
17
17
|
"analyzer": ["config_sanity_checks", "get_dd_analyzer", "ServiceFactory", "update_cfg_from_defaults"],
|
|
18
18
|
"eval": [
|
|
@@ -340,7 +340,7 @@ class D2FrcnnDetector(D2FrcnnDetectorMixin):
|
|
|
340
340
|
path_weights: PathLikeOrStr,
|
|
341
341
|
categories: Mapping[int, TypeOrStr],
|
|
342
342
|
config_overwrite: Optional[list[str]] = None,
|
|
343
|
-
device: Optional[Union[Literal["cpu", "cuda"], torch.device]] = None,
|
|
343
|
+
device: Optional[Union[Literal["cpu", "cuda", "mps"], torch.device]] = None,
|
|
344
344
|
filter_categories: Optional[Sequence[TypeOrStr]] = None,
|
|
345
345
|
):
|
|
346
346
|
"""
|
|
@@ -369,7 +369,10 @@ class D2FrcnnDetector(D2FrcnnDetectorMixin):
|
|
|
369
369
|
|
|
370
370
|
config_overwrite = config_overwrite if config_overwrite else []
|
|
371
371
|
self.config_overwrite = config_overwrite
|
|
372
|
-
|
|
372
|
+
requested = get_torch_device(device)
|
|
373
|
+
# We need to change device for mps otherwise inference takes forever:
|
|
374
|
+
# https://github.com/facebookresearch/detectron2/issues/4888
|
|
375
|
+
self.device = torch.device("cpu") if requested.type == "mps" else requested
|
|
373
376
|
|
|
374
377
|
d2_conf_list = self._get_d2_config_list(path_weights, config_overwrite)
|
|
375
378
|
self.cfg = self._set_config(path_yaml, d2_conf_list, self.device)
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/conf_tesseract.yaml
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
{deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/dependency_links.txt
RENAMED
|
File without changes
|
|
File without changes
|
|
File without changes
|