Skip to content

Instantly share code, notes, and snippets.

@appliedml42

appliedml42/quickdraw_example.py

Last active November 20, 2022 17:33
Show Gist options
  • Save appliedml42/90da9aaec00846da1809c9014e2a1dd6 to your computer and use it in GitHub Desktop.
Save appliedml42/90da9aaec00846da1809c9014e2a1dd6 to your computer and use it in GitHub Desktop.
Download ZIP
Using QuickDraw Dataset
Raw
quickdraw_example.py
from am42_fsd.data.quickdraw import QuickDraw
from torch.utils.data import DataLoader
from tqdm.auto import tqdm
quickdraw = QuickDraw("path to numpy_bitmap folder")
dataloader = DataLoader(quickdraw, shuffle=True, batch_size=64, num_workers=4)
for images, cat_id in tqdm(dataloader):
pass
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment