Torchvision's v2 transforms support for transform images and bounding boxes similar to how we implement via ExTransform: https://github.com/IntelLabs/MART/blob/20d20789548c232e8d9a13079c2cef222fb9889b/mart/transforms/extended.py#L35 I think we should just standardize on Torchvision's implementation. https://pytorch.org/vision/main/auto_examples/plot_transforms_v2.html
Torchvision's v2 transforms support for transform images and bounding boxes similar to how we implement via ExTransform:
MART/mart/transforms/extended.py
Line 35 in 20d2078
I think we should just standardize on Torchvision's implementation.
https://pytorch.org/vision/main/auto_examples/plot_transforms_v2.html