maite-datasets 0.0.5__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.
- maite_datasets/__init__.py +2 -6
- maite_datasets/_base.py +169 -51
- maite_datasets/_builder.py +46 -55
- maite_datasets/_collate.py +2 -3
- maite_datasets/{_reader/_base.py → _reader.py} +62 -36
- maite_datasets/_validate.py +4 -2
- maite_datasets/image_classification/_cifar10.py +12 -7
- maite_datasets/image_classification/_mnist.py +15 -10
- maite_datasets/image_classification/_ships.py +12 -8
- maite_datasets/object_detection/__init__.py +4 -7
- maite_datasets/object_detection/_antiuav.py +11 -8
- maite_datasets/{_reader → object_detection}/_coco.py +29 -27
- maite_datasets/object_detection/_milco.py +11 -9
- maite_datasets/object_detection/_seadrone.py +11 -9
- maite_datasets/object_detection/_voc.py +11 -13
- maite_datasets/{_reader → object_detection}/_yolo.py +26 -21
- maite_datasets/protocols.py +23 -0
- maite_datasets/wrappers/__init__.py +8 -0
- maite_datasets/wrappers/_torch.py +111 -0
- {maite_datasets-0.0.5.dist-info → maite_datasets-0.0.6.dist-info}/METADATA +56 -3
- maite_datasets-0.0.6.dist-info/RECORD +26 -0
- maite_datasets/_mixin/__init__.py +0 -0
- maite_datasets/_mixin/_numpy.py +0 -28
- maite_datasets/_mixin/_torch.py +0 -28
- maite_datasets/_protocols.py +0 -217
- maite_datasets/_reader/__init__.py +0 -6
- maite_datasets/_reader/_factory.py +0 -64
- maite_datasets/_types.py +0 -50
- maite_datasets/object_detection/_voc_torch.py +0 -65
- maite_datasets-0.0.5.dist-info/RECORD +0 -31
- {maite_datasets-0.0.5.dist-info → maite_datasets-0.0.6.dist-info}/WHEEL +0 -0
- {maite_datasets-0.0.5.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.5.dist-info/METADATA,sha256=jUunRXY6i0dwnX8JWTYTzEJgD2wOapp3I2xdhjuZ6kI,3747
|
29
|
-
maite_datasets-0.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
30
|
-
maite_datasets-0.0.5.dist-info/licenses/LICENSE,sha256=6h3J3R-ajGHh_isDSftzS5_jJjB9HH4TaI0vU-VscaY,1082
|
31
|
-
maite_datasets-0.0.5.dist-info/RECORD,,
|
File without changes
|
File without changes
|