mlcommons / GaNDLF

A generalizable application framework for segmentation, regression, and classification using PyTorch
https://gandlf.org
Apache License 2.0
150 stars 78 forks source link

[FEATURE] add function to retrieve augmentations from config #888

Closed szmazurek closed 2 weeks ago

szmazurek commented 3 weeks ago

Is your feature request related to a problem? Please describe.

N/A

Describe the solution you'd like

Add a function that would handle getting augmentation transforms from parameters. Would also be useful in future gandalf-dependent packages.

Describe alternatives you've considered

N/A

sarthakpati commented 2 weeks ago

Done with #889