-
Notifications
You must be signed in to change notification settings - Fork 407
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
Convert all index transforms to Kornia #999
Conversation
Can't get the dimensions to work in the transforms tutorial, please help. |
The only thing I can get to work is to match the size dimension of the input.
|
Well, all the dims work now, but the plotted image is all black. Not sure what changed to cause that. |
Heads up that running |
7005321
to
7ec4d93
Compare
* Convert all index transforms to Kornia * Fix tests too * Fix docs build * Update tutorial * Update transforms tutorial * Kornia augs add new batch dimension * Expand dims * Fix one more set of transforms * Escape quotes * Remove unneeded transform * Kornia transforms require float images * Fix device problems Co-authored-by: Caleb Robinson <[email protected]>
This PR converts all of our index transforms to be valid Kornia augmentations. These transforms can now be used with AugmentationSequential just like all of the other transforms we use.