ColoredRows

class torchelie.datasets.ColoredRows(*size, transform=None)

A dataset of precedurally generated images of rows randomly colorized.

Parameters

• *size (int) – size of images

• transform (transforms or None) – the image transforms to apply to the generated pictures