ColoredColumns

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

A dataset of precedurally generated images of columns randomly colorized.

Parameters

• *size (int) – size of images

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