maite-datasets 0.0.5__py3-none-any.whl → 0.0.7__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 (35) 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/adapters/__init__.py +3 -0
  8. maite_datasets/adapters/_huggingface.py +391 -0
  9. maite_datasets/image_classification/_cifar10.py +12 -7
  10. maite_datasets/image_classification/_mnist.py +15 -10
  11. maite_datasets/image_classification/_ships.py +12 -8
  12. maite_datasets/object_detection/__init__.py +4 -7
  13. maite_datasets/object_detection/_antiuav.py +11 -8
  14. maite_datasets/{_reader → object_detection}/_coco.py +29 -27
  15. maite_datasets/object_detection/_milco.py +11 -9
  16. maite_datasets/object_detection/_seadrone.py +11 -9
  17. maite_datasets/object_detection/_voc.py +11 -13
  18. maite_datasets/{_reader → object_detection}/_yolo.py +26 -21
  19. maite_datasets/protocols.py +94 -0
  20. maite_datasets/wrappers/__init__.py +8 -0
  21. maite_datasets/wrappers/_torch.py +109 -0
  22. maite_datasets-0.0.7.dist-info/METADATA +181 -0
  23. maite_datasets-0.0.7.dist-info/RECORD +28 -0
  24. maite_datasets/_mixin/__init__.py +0 -0
  25. maite_datasets/_mixin/_numpy.py +0 -28
  26. maite_datasets/_mixin/_torch.py +0 -28
  27. maite_datasets/_protocols.py +0 -217
  28. maite_datasets/_reader/__init__.py +0 -6
  29. maite_datasets/_reader/_factory.py +0 -64
  30. maite_datasets/_types.py +0 -50
  31. maite_datasets/object_detection/_voc_torch.py +0 -65
  32. maite_datasets-0.0.5.dist-info/METADATA +0 -91
  33. maite_datasets-0.0.5.dist-info/RECORD +0 -31
  34. {maite_datasets-0.0.5.dist-info → maite_datasets-0.0.7.dist-info}/WHEEL +0 -0
  35. {maite_datasets-0.0.5.dist-info → maite_datasets-0.0.7.dist-info}/licenses/LICENSE +0 -0
@@ -1,91 +0,0 @@
1
- Metadata-Version: 2.4
2
- Name: maite-datasets
3
- Version: 0.0.5
4
- Summary: A collection of Image Classification and Object Detection task datasets conforming to the MAITE protocol.
5
- Author-email: Andrew Weng <andrew.weng@ariacoustics.com>, Ryan Wood <ryan.wood@ariacoustics.com>, Shaun Jullens <shaun.jullens@ariacoustics.com>
6
- License-Expression: MIT
7
- License-File: LICENSE
8
- Classifier: Development Status :: 4 - Beta
9
- Classifier: Framework :: Pytest
10
- Classifier: License :: OSI Approved :: MIT License
11
- Classifier: Operating System :: OS Independent
12
- Classifier: Programming Language :: Python :: 3 :: Only
13
- Classifier: Programming Language :: Python :: 3.9
14
- Classifier: Programming Language :: Python :: 3.10
15
- Classifier: Programming Language :: Python :: 3.11
16
- Classifier: Programming Language :: Python :: 3.12
17
- Requires-Python: >=3.9
18
- Requires-Dist: defusedxml>=0.7.1
19
- Requires-Dist: numpy>=1.24.2
20
- Requires-Dist: pillow>=10.3.0
21
- Requires-Dist: requests>=2.32.3
22
- Requires-Dist: typing-extensions>=4.12
23
- Provides-Extra: tqdm
24
- Requires-Dist: tqdm>=4.66; extra == 'tqdm'
25
- Description-Content-Type: text/markdown
26
-
27
- # MAITE Datasets
28
-
29
- MAITE Datasets are a collection of public datasets wrapped in a [MAITE](https://mit-ll-ai-technology.github.io/maite/) compliant format.
30
-
31
- ## Installation
32
-
33
- To install and use `maite-datasets` you can use pip:
34
-
35
- ```bash
36
- pip install maite-datasets
37
- ```
38
-
39
- For status bar indicators when downloading, you can include the extra `tqdm` when installing:
40
-
41
- ```bash
42
- pip install maite-datasets[tqdm]
43
- ```
44
-
45
- ## Available Datasets
46
-
47
- | Task | Dataset | Description |
48
- |----------------|------------------|---------------------------------------------------------------------|
49
- | Classification | CIFAR10 | [CIFAR10](https://www.cs.toronto.edu/~kriz/cifar.html) dataset. |
50
- | Classification | MNIST | A dataset of hand-written digits. |
51
- | Classification | Ships | A dataset that focuses on identifying ships from satellite images. |
52
- | Detection | AntiUAVDetection | A UAV detection dataset in natural images with varying backgrounds. |
53
- | Detection | MILCO | A side-scan sonar dataset focused on mine-like object detection. |
54
- | Detection | Seadrone | A UAV dataset focused on open water object detection. |
55
- | Detection | VOCDetection | [Pascal VOC](http://host.robots.ox.ac.uk/pascal/VOC/) dataset. |
56
-
57
- ## Usage
58
-
59
- Here is an example of how to import MNIST for usage with your workflow.
60
-
61
- ```python
62
- >>> from maite_datasets.image_classification import MNIST
63
-
64
- >>> mnist = MNIST(root="data", download=True)
65
- >>> print(mnist)
66
- MNIST Dataset
67
- -------------
68
- Corruption: None
69
- Transforms: []
70
- Image_set: train
71
- Metadata: {'id': 'MNIST_train', 'index2label': {0: 'zero', 1: 'one', 2: 'two', 3: 'three', 4: 'four', 5: 'five', 6: 'six', 7: 'seven', 8: 'eight', 9: 'nine'}, 'split': 'train'}
72
- Path: /home/user/maite-datasets/data/mnist
73
- Size: 60000
74
-
75
- >>> print("tuple("+", ".join([str(type(t)) for t in mnist[0]])+")")
76
- tuple(<class 'numpy.ndarray'>, <class 'numpy.ndarray'>, <class 'dict'>)
77
- ```
78
-
79
- ## Additional Information
80
-
81
- For more information on the MAITE protocol, check out their [documentation](https://mit-ll-ai-technology.github.io/maite/).
82
-
83
- ## Acknowledgement
84
-
85
- ### CDAO Funding Acknowledgement
86
-
87
- This material is based upon work supported by the Chief Digital and Artificial
88
- Intelligence Office under Contract No. W519TC-23-9-2033. The views and
89
- conclusions contained herein are those of the author(s) and should not be
90
- interpreted as necessarily representing the official policies or endorsements,
91
- either expressed or implied, of the U.S. Government.
@@ -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,,