Как реализовать кастомный Dataset?

Question

Accepted Answer

Класс с `__len__` и `__getitem__`: по индексу читает sample, делает transforms и возвращает тензоры/словарь полей.

## Разбор

- Тяжёлое I/O лучше в workers DataLoader.
- Единый контракт ключей упрощает collate.
- Кэш в RAM — только если данные помещаются.
- Ошибки индексации ловят unit-тестами длины и типов.

## Пример

```python
class ImgDS(Dataset):
    def __init__(self, paths, labels, tfm):
        self.paths, self.labels, self.tfm = paths, labels, tfm
    def __len__(self): return len(self.paths)
    def __getitem__(self, i):
        return self.tfm(load(self.paths[i])), self.labels[i]
```

## Итог

Кастомный Dataset = индекс → преобразованный sample.