FluxML / FluxML/DataAugmentation.jl

Support for GPU-accelerated affine transformations

Open
#48 3 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Julia
Stars
46
Forks
18
PR merge metrics
No merged PRs in 30d

Description

Affine transformations on CPU are already pretty fast, but GPU-accelerated transformations are useful in CPU-constrained environments like Google Colab where a GPU is available but only 2 mediocre CPUs which can lead to a bottleneck in the data pipeline.

Some open questions:

- Would the existing transforms work out-of-the box on `CuArray`s (i.e. does `ImageTransformations.warp[!]` work on `CuArray`s) ? What about for masks, i.e. integer arrays? Otherwise, what would be necessary to implement that?
- Performance-wise, do the transforms need to be applied to a whole batch at once or is it as fast to apply them to samples individually? Former would require first resizing images to the same size and wrapping in a `Batch` wrapper item.

Can someone who has experience with image transformations on GPU chime in?

@jsamaroo

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.