task_categories:other auto-generated image-classification region:us

---
task_categories:
- other
task_ids:
- other-image-classification
- image-classification
tags:
- auto-generated
- image-classification
---

# nateraw/auto-cats-and-dogs

Image Classification Dataset

## Usage

```python
from PIL import Image
from datasets import load_dataset

def pil_loader(path: str):
    with open(path, 'rb') as f:
        im = Image.open(f)
        return im.convert('RGB')

def image_loader(example_batch):
    example_batch['image'] = [
        pil_loader(f) for f in example_batch['file']
    ]
    return example_batch


ds = load_dataset('nateraw/auto-cats-and-dogs')
ds = ds.with_transform(image_loader)
```

