atomai icon indicating copy to clipboard operation
atomai copied to clipboard

Custom dataloaders for AtomAI models

Open ziatdinovmax opened this issue 1 year ago • 1 comments

Passing a custom data loader in AtomAI models can be a useful feature. This would allow using e.g. Kornia data augmentation pipelines. It could look like this

segmodel = aoi.models.Segmentor(nb_classes=3)
segmodel.fit(train_loader=custom_train_loader, test_loader=custom_test_loader). # instead of passing X_train, y_train, X_test, y_test

It will also be helpful to have a utility function that builds a dataloader with Kornia data augmentation functions.

ziatdinovmax avatar Mar 29 '23 21:03 ziatdinovmax

I will add the custom data loader functionality, starting with the Segmentor class.

ahoust17 avatar Feb 02 '24 21:02 ahoust17