download_tiny_imagenet.py

import torch
import torchvision
from pathlib import Path
import zipfile

path = Path('./data/tiny-imagenet-200.zip')
extracted_path = path.parent / path.stem


if extracted_path.exists():
    print("path already exists")
else:
    url = "http://cs231n.stanford.edu/tiny-imagenet-200.zip"
    md5 = "90528d7ca1a48142e341f4ef8d21d0de"  # set to None if you don't know the correct md5
    torchvision.datasets.utils.download_url(url, path.parent, md5=md5)

    if md5 is None:
        print("md5", torchvision.datasets.utils.calculate_md5(path))
    
    with zipfile.ZipFile(path) as z:
        z.extractall(path.parent)