Library that provides a QuickDraw dataset using the Pytorch API.
Project description
Quick, Torch!
Quick, Torch! is a simple package that provides a "Quick, Draw!" dataset using the abstract class VisionDataset
, provided by torchvision
API. This package mirrors the MNIST dataset provided in torchvision.
You can install this package with
pip install quick-torch --upgrade
Example
Here are a simple example of usage:
from quick_torch import QuickDraw
import torchvision.transforms as T
ds = QuickDraw(
root="dataset",
categories="face",
download=True,
transform=T.Resize((128, 128))
)
print(f"{len(ds) = }")
first_data = ds[0]
first_data
>>> Downloading https://storage.googleapis.com/quickdraw_dataset/full/numpy_bitmap/face.npy
>>>
>>> len(ds) = 161666
>>> (<PIL.Image.Image image mode=L size=128x128>, 108)
For more examples, please refer to the notebook example.ipynb
References
This work was mainly inspired by the following repositories:
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
quick_torch-1.0.4.tar.gz
(11.0 kB
view hashes)
Built Distribution
Close
Hashes for quick_torch-1.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 30ec835623b54f68ccc069d09df2e900e2179212cbc3a2d5fb44f539aa20e158 |
|
MD5 | c20ce25c27feea2bc798bbed7bdb11a8 |
|
BLAKE2b-256 | a379cb76a34855b928b9df417a572054818f408931e2d795d325461f2c0d4728 |