maite-datasets 0.0.4a0__py3-none-any.whl → 0.0.6__py3-none-any.whl

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 (32) hide show
  1. maite_datasets/__init__.py +2 -6
  2. maite_datasets/_base.py +169 -51
  3. maite_datasets/_builder.py +46 -55
  4. maite_datasets/_collate.py +2 -3
  5. maite_datasets/{_reader/_base.py → _reader.py} +62 -36
  6. maite_datasets/_validate.py +4 -2
  7. maite_datasets/image_classification/_cifar10.py +12 -7
  8. maite_datasets/image_classification/_mnist.py +15 -10
  9. maite_datasets/image_classification/_ships.py +12 -8
  10. maite_datasets/object_detection/__init__.py +4 -7
  11. maite_datasets/object_detection/_antiuav.py +11 -8
  12. maite_datasets/{_reader → object_detection}/_coco.py +29 -27
  13. maite_datasets/object_detection/_milco.py +11 -9
  14. maite_datasets/object_detection/_seadrone.py +11 -9
  15. maite_datasets/object_detection/_voc.py +11 -13
  16. maite_datasets/{_reader → object_detection}/_yolo.py +26 -21
  17. maite_datasets/protocols.py +23 -0
  18. maite_datasets/wrappers/__init__.py +8 -0
  19. maite_datasets/wrappers/_torch.py +111 -0
  20. {maite_datasets-0.0.4a0.dist-info → maite_datasets-0.0.6.dist-info}/METADATA +56 -3
  21. maite_datasets-0.0.6.dist-info/RECORD +26 -0
  22. maite_datasets/_mixin/__init__.py +0 -0
  23. maite_datasets/_mixin/_numpy.py +0 -28
  24. maite_datasets/_mixin/_torch.py +0 -28
  25. maite_datasets/_protocols.py +0 -217
  26. maite_datasets/_reader/__init__.py +0 -6
  27. maite_datasets/_reader/_factory.py +0 -64
  28. maite_datasets/_types.py +0 -50
  29. maite_datasets/object_detection/_voc_torch.py +0 -65
  30. maite_datasets-0.0.4a0.dist-info/RECORD +0 -31
  31. {maite_datasets-0.0.4a0.dist-info → maite_datasets-0.0.6.dist-info}/WHEEL +0 -0
  32. {maite_datasets-0.0.4a0.dist-info → maite_datasets-0.0.6.dist-info}/licenses/LICENSE +0 -0
@@ -1,65 +0,0 @@
1
- from __future__ import annotations
2
-
3
- __all__ = []
4
-
5
- from torch import Tensor
6
-
7
- from maite_datasets._base import BaseODDataset
8
- from maite_datasets._types import ObjectDetectionTarget
9
- from maite_datasets._mixin._torch import BaseDatasetTorchMixin
10
- from maite_datasets.object_detection._voc import BaseVOCDataset
11
-
12
-
13
- class VOCDetectionTorch(
14
- BaseVOCDataset[Tensor, ObjectDetectionTarget[Tensor]],
15
- BaseODDataset[Tensor, list[str], str],
16
- BaseDatasetTorchMixin,
17
- ):
18
- """
19
- `Pascal VOC <http://host.robots.ox.ac.uk/pascal/VOC/>`_ Detection Dataset as PyTorch tensors.
20
-
21
- Parameters
22
- ----------
23
- root : str or pathlib.Path
24
- Because of the structure of the PASCAL VOC datasets, the root needs to be one of 4 folders.
25
- 1) Directory containing the year of the **already downloaded** dataset (i.e. .../VOCdevkit/VOC2012 <-)
26
- 2) Directory to the VOCdevkit folder of the **already downloaded** dataset (i.e. .../VOCdevkit <- /VOC2012)
27
- 3) Directory to the folder one level up from the VOCdevkit folder,
28
- data **may** or **may not** be already downloaded (i.e. ... <- /VOCdevkit/VOC2012)
29
- 4) Directory to where you would like the dataset to be downloaded
30
- image_set : "train", "val", "test", or "base", default "train"
31
- If "test", then dataset year must be "2007" or "2012". Note that the 2012 test set does not contain annotations.
32
- If "base", then the combined dataset of "train" and "val" is returned.
33
- year : "2007", "2008", "2009", "2010", "2011" or "2012", default "2012"
34
- The dataset year.
35
- transforms : Transform, Sequence[Transform] or None, default None
36
- Transform(s) to apply to the data.
37
- download : bool, default False
38
- If True, downloads the dataset from the internet and puts it in root directory.
39
- Class checks to see if data is already downloaded to ensure it does not create a duplicate download.
40
- verbose : bool, default False
41
- If True, outputs print statements.
42
-
43
- Attributes
44
- ----------
45
- path : pathlib.Path
46
- Location of the folder containing the data.
47
- year : "2007", "2008", "2009", "2010", "2011" or "2012"
48
- The selected dataset year.
49
- image_set : "train", "val", "test" or "base"
50
- The selected image set from the dataset.
51
- index2label : dict[int, str]
52
- Dictionary which translates from class integers to the associated class strings.
53
- label2index : dict[str, int]
54
- Dictionary which translates from class strings to the associated class integers.
55
- metadata : DatasetMetadata
56
- Typed dictionary containing dataset metadata, such as `id` which returns the dataset class name.
57
- transforms : Sequence[Transform]
58
- The transforms to be applied to the data.
59
- size : int
60
- The size of the dataset.
61
-
62
- Note
63
- ----
64
- Data License: `Flickr Terms of Use <http://www.flickr.com/terms.gne?legacy=1>`_
65
- """
@@ -1,31 +0,0 @@
1
- maite_datasets/__init__.py,sha256=53LW5bHMAr4uD6w2bvrPxgtROUIzaE-3LR6TR0dDucs,746
2
- maite_datasets/_base.py,sha256=BiWB_xvL4AtV0jxVjzpcZHuRTb52dTD0CQtu08DzoXA,8195
3
- maite_datasets/_builder.py,sha256=yESeNf4MBm4oSPKse6jSUt0XZrxkSU2etdK82wXiiUw,10071
4
- maite_datasets/_collate.py,sha256=-XuKeeMmOnSB0RgQbz8BjsoqQar9Tsf_qALZxijQ498,4063
5
- maite_datasets/_fileio.py,sha256=7S-hF3xU60AdcsPsfYR7rjbeGZUlv3JjGEZhGJOxGYU,5622
6
- maite_datasets/_protocols.py,sha256=aWrnUM1stZ9VInkBEynod_OdYq2ORSpew7yoF-Zeuig,5247
7
- maite_datasets/_types.py,sha256=S5DMyiUrkUjV9uM0ysKqxVoi7z5P7B3EPiLI4Fyq9Jc,1147
8
- maite_datasets/_validate.py,sha256=sP-5lYXkmkiTadJcy_LtEMiZ0m82xR0yELoxWORrZDQ,6904
9
- maite_datasets/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
- maite_datasets/_mixin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
- maite_datasets/_mixin/_numpy.py,sha256=GEuRyeprH-STh-_zktAp0Tg6NNyMdh1ThyhjW558NOo,860
12
- maite_datasets/_mixin/_torch.py,sha256=pkN2vMNsDk_h5wnD5899zIHsPtEADbGfmRyI5CdGonI,827
13
- maite_datasets/_reader/__init__.py,sha256=VzrVOsmztPJV83um8tY5qdqU-HEPP15RlLClGbxTFlQ,164
14
- maite_datasets/_reader/_base.py,sha256=nQc3wqI02N6sC9Rk98BT6vFJa5VP9pAZjKhpyqXfW0o,4299
15
- maite_datasets/_reader/_coco.py,sha256=sCrOixqOuqgHpee3NjDTt_bhbxgKGean2mn2Py9mh5k,10127
16
- maite_datasets/_reader/_factory.py,sha256=cI3Cw1yWj4hK2gn6N5bugXzGMcNwcCEkJ4AoynwOZvI,2222
17
- maite_datasets/_reader/_yolo.py,sha256=29eQMzBuJpaKFMfC0Pu_RvrGnqbqQxY0XZmeSPaYArg,11626
18
- maite_datasets/image_classification/__init__.py,sha256=pcZojkdsiMoLgY4mKjoQY6WyEwiGYHxNrAGpnvn3zsY,308
19
- maite_datasets/image_classification/_cifar10.py,sha256=w7BPGZzUV1gXFoYRgxa6VOqKn1EgQi3x1rrA4nEUbeI,8470
20
- maite_datasets/image_classification/_mnist.py,sha256=6xDWY4qbY1hlcUZKvVZeQMvYbF0vLtaVzOuQUKJkcJU,8248
21
- maite_datasets/image_classification/_ships.py,sha256=_fkm4iu6xuvfRuivgIS8S3CYnQOgghi9Kc0Riz1Dr8g,5187
22
- maite_datasets/object_detection/__init__.py,sha256=NE8apy2C0kTg_Ng_M15U21ZW66WC_LWezmdG8vk2WHM,590
23
- maite_datasets/object_detection/_antiuav.py,sha256=2xFOOCT2aujkD6T9LHJfUd02zyTsoNlLZ_rxqztUBP0,8333
24
- maite_datasets/object_detection/_milco.py,sha256=KEU4JFvCxfyMAb4RFMnxTMk_MggdEAV8y4LU-kjN3lE,7997
25
- maite_datasets/object_detection/_seadrone.py,sha256=w_pSojLzgwdKrUSxaz8r7dPJVKGND6JSYl0S_BKOLH0,271282
26
- maite_datasets/object_detection/_voc.py,sha256=VuokKaOzI1wSfgG5DC7ufMbRDlG-b6Se3hg4eQzNQbE,19731
27
- maite_datasets/object_detection/_voc_torch.py,sha256=bjeawnNit7Llcf_cZY_9lcJYoUoAU-Wen6MMT-7QX3k,2917
28
- maite_datasets-0.0.4a0.dist-info/METADATA,sha256=P0IvwtwKWspvO8Tl_V1PuPfH_XbIf0E-0gj5Qxvgrug,3749
29
- maite_datasets-0.0.4a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
30
- maite_datasets-0.0.4a0.dist-info/licenses/LICENSE,sha256=6h3J3R-ajGHh_isDSftzS5_jJjB9HH4TaI0vU-VscaY,1082
31
- maite_datasets-0.0.4a0.dist-info/RECORD,,