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.
Files changed (58) hide show
  1. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/PKG-INFO +1 -1
  2. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/pyproject.toml +1 -1
  3. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/__init__.py +1 -1
  4. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/d2detect.py +5 -2
  5. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/PKG-INFO +1 -1
  6. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/README.md +0 -0
  7. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/setup.cfg +0 -0
  8. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/__init__.py +0 -0
  9. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/config.py +0 -0
  10. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/dd.py +0 -0
  11. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/analyzer/factory.py +0 -0
  12. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/__init__.py +0 -0
  13. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/conf_dd_one.yaml +0 -0
  14. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/conf_tesseract.yaml +0 -0
  15. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/configs/profiles.jsonl +0 -0
  16. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/__init__.py +0 -0
  17. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/accmetric.py +0 -0
  18. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/base.py +0 -0
  19. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/cocometric.py +0 -0
  20. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/eval.py +0 -0
  21. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/registry.py +0 -0
  22. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/eval/tedsmetric.py +0 -0
  23. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/__init__.py +0 -0
  24. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/base.py +0 -0
  25. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/deskew.py +0 -0
  26. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/doctrocr.py +0 -0
  27. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/hfdetr.py +0 -0
  28. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/hflayoutlm.py +0 -0
  29. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/hflm.py +0 -0
  30. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/model.py +0 -0
  31. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/pdftext.py +0 -0
  32. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/tessocr.py +0 -0
  33. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/extern/texocr.py +0 -0
  34. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/__init__.py +0 -0
  35. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/anngen.py +0 -0
  36. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/base.py +0 -0
  37. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/common.py +0 -0
  38. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/concurrency.py +0 -0
  39. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/doctectionpipe.py +0 -0
  40. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/language.py +0 -0
  41. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/layout.py +0 -0
  42. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/lm.py +0 -0
  43. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/order.py +0 -0
  44. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/refine.py +0 -0
  45. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/registry.py +0 -0
  46. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/segment.py +0 -0
  47. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/sub_layout.py +0 -0
  48. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/text.py +0 -0
  49. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/pipe/transform.py +0 -0
  50. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/py.typed +0 -0
  51. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/__init__.py +0 -0
  52. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/d2_frcnn_train.py +0 -0
  53. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/hf_detr_train.py +0 -0
  54. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection/train/hf_layoutlm_train.py +0 -0
  55. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/SOURCES.txt +0 -0
  56. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/dependency_links.txt +0 -0
  57. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/requires.txt +0 -0
  58. {deepdoctection-1.0.2 → deepdoctection-1.0.4}/src/deepdoctection.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: deepdoctection
3
- Version: 1.0.2
3
+ Version: 1.0.4
4
4
  Summary: Repository for Document AI - server/inference core package
5
5
  Author: Dr. Janis Meyer
6
6
  License: Apache License 2.0
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "deepdoctection"
7
- version = "1.0.2"
7
+ version = "1.0.4"
8
8
  authors = [
9
9
  {name = "Dr. Janis Meyer"}
10
10
  ]
@@ -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.2"
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
- self.device = get_torch_device(device)
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)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: deepdoctection
3
- Version: 1.0.2
3
+ Version: 1.0.4
4
4
  Summary: Repository for Document AI - server/inference core package
5
5
  Author: Dr. Janis Meyer
6
6
  License: Apache License 2.0
File without changes
File without changes