search

juliendenize / torchaug

Library to perform efficient vision data augmentations for CPU/GPU per-sample/batched data.
https://torchaug.readthedocs.io/en/latest/
Other
24 stars 2 forks source link

Refractor on torchvision v2 #18

Closed juliendenize closed 7 months ago

juliendenize commented 7 months ago

Torchvision released the v2 transform API.

https://pytorch.org/vision/0.17/transforms.html#v1-or-v2-which-one-should-i-use

Torchaug will not only support this new API but be refractor on top of it to support enhancements and augmentations on bounding boxes and masks.