Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Custom dataloaders for AtomAI models #71

Open
ziatdinovmax opened this issue Mar 29, 2023 · 1 comment
Open

Custom dataloaders for AtomAI models #71

ziatdinovmax opened this issue Mar 29, 2023 · 1 comment

Comments

@ziatdinovmax
Copy link
Collaborator

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.

@ahoust17
Copy link
Contributor

ahoust17 commented Feb 2, 2024

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants