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

Add nested tensor #58

Closed juliendenize closed 3 months ago

juliendenize commented 3 months ago

Add the TANestedTensors that allow a list of tensors to be passed for transforms to adress https://github.com/juliendenize/torchaug/issues/56

Essentially the transforms apply a for loop to the tensors. Several helpers are provided to load nested tensors and convert them.

The idea is to propose a way to have nested tensors on GPU that can be resized and then converte to batches for training or to list for inference.

codecov[bot] commented 3 months ago

Codecov Report

Attention: Patch coverage is 79.68750% with 91 lines in your changes missing coverage. Please review.

Project coverage is 87.33%. Comparing base (f0b97b6) to head (11d0f42).

Files Patch % Lines
torchaug/ta_tensors/nested/_ta_nested_tensors.py 75.28% 65 Missing :warning:
torchaug/data/dataloader/_collate.py 46.42% 15 Missing :warning:
torchaug/transforms/_transform.py 83.33% 9 Missing :warning:
torchaug/transforms/functional/_type_conversion.py 81.81% 2 Missing :warning:
Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #58 +/- ## ========================================== - Coverage 88.19% 87.33% -0.87% ========================================== Files 45 54 +9 Lines 3998 4420 +422 ========================================== + Hits 3526 3860 +334 - Misses 472 560 +88 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.