A collection of popular datasets for deep learning.
For the latest source, discussion, etc, please visit the GitHub repository
dbcollection is a library for downloading/parsing/managing datasets via simple methods. It was built from the ground up to be cross-platform (Windows, Linux, MacOS) and cross-language (Python, Lua, Matlab, etc.). This is achieved by using the popular HDF5 file format to store (meta)data of manually parsed datasets and the power of Python for scripting. By doing so, this library can target any platform that supports Python and any language that has bindings for HDF5.
This package allows to easily manage and load datasets by using HDF5 files to store metadata. By storing all the necessary metadata to disk, managing either big or small datasets has an equal or very similar impact on the system’s resource usage. Also, once a dataset is setup, it is setup forever! This means users can reuse any previously set dataset as many times as needed without having to set it each time they are used.
dbcollection allows users to focus on more important tasks like prototyping new models or testing them in different datasets without having to incur the loss of spending time managing datasets or creating/modyfing scripts to load/fetch data by taking advantage of the work of the community that shared these resources.
Here are some of key features dbcollection provides:
- Simple API to load/download/setup/manage datasets.
- Simple API to fetch data from a dataset.
- Store and pull data from disk or from memory, you choose!
- Datasets only need to be set/processed once, so next time you use it it will load instantly!
- Cross-platform (Windows, Linux, MacOs).
- Cross-language (Python, Lua/Torch7, Matlab).
- Easily extensible to other languages that support HDF5 files format.
- Concurrent/parallel data access thanks to HDF5.
- Contains a diverse (and growing!) list of popular datasets for machine-, deep-learning tasks (object detection, action recognition, human pose estimation, etc.)
Installing dbcollection using pip is simple. For that purpose, simply do the following command:
$ pip install dbcollection
To install dbcollection from source you need to do the following setps:
- Clone the repo to your hard drive:
$ git clone --recursive https://github.com/dbcollection/dbcollection
- cd to the dbcollection folder and do the command
$ python setup.py install
Using the module is pretty straight-forward. To import it just do:
>>> import dbcollection as dbc
To load a dataset, you only need to use a single method that returns a data loader object which can then be used to fetch data from.
>>> mnist = dbc.load('mnist')
This data loader object contains information about the dataset’s name, task, data, cache paths, set splits, and some methods for querying and loading data from the HDF5 metadata file.
For example, if you want to know how the data is structured inside the metadata file, you can simply do the following:
>>> mnist.info() > Set: test - classes, shape = (10, 2), dtype = uint8 - images, shape = (10000, 28, 28), dtype = uint8, (in 'object_ids', position = 0) - labels, shape = (10000,), dtype = uint8, (in 'object_ids', position = 1) - object_fields, shape = (2, 7), dtype = uint8 - object_ids, shape = (10000, 2), dtype = uint8 (Pre-ordered lists) - list_images_per_class, shape = (10, 1135), dtype = int32 > Set: train - classes, shape = (10, 2), dtype = uint8 - images, shape = (60000, 28, 28), dtype = uint8, (in 'object_ids', position = 0) - labels, shape = (60000,), dtype = uint8, (in 'object_ids', position = 1) - object_fields, shape = (2, 7), dtype = uint8 - object_ids, shape = (60000, 2), dtype = uint8 (Pre-ordered lists) - list_images_per_class, shape = (10, 6742), dtype = int32
To fetch data samples from a field, its is as easy as calling a method with the set and field names and the row id(s) you want to select. For example, to retrieve the 10 first images all you need to do is the following:
>>> imgs = mnist.get('train', 'images', range(10)) >>> imgs.shape (10, 28, 28)
Note: For more information about using this module, please check the documentation or the available notebooks for guidance.
For a more pratical introduction to dbcollection’s module for managing datasets and fetching data, there are some python notebooks available in the notebooks/ folder for a more hands-on tutorial on how to use this package.
The package documentation is hosted on Read The Docs.
It provides a more detailed guide on how to use this package as well as additional information that you might find relevant about this project.
All contributions, bug reports, bug fixes, documentation improvements, enhancements and ideas are welcome. If you would like to see additional languages being supported, please consider contributing to the project.
If you are interested in fixing issues and contributing directly to the code base, please see the document How to Contribute.
Release history Release notifications
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size & hash SHA256 hash help||File type||Python version||Upload date|
|dbcollection-0.2.6-py2.py3-none-any.whl (11.8 MB) Copy SHA256 hash SHA256||Wheel||3.6|
|dbcollection-0.2.6.tar.gz (11.6 MB) Copy SHA256 hash SHA256||Source||None|