-
Hi guys, I was wondering if there's an existent equivalent to PyTorch's interpolate already in JAX (maybe .scipy?). I'm working on porting Torchvision's implementations of segmentation models to JAX/Flax. Let me know what I could use for this. Many thanks! |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
Take a look at |
Beta Was this translation helpful? Give feedback.
Take a look at
jax.image.resize
andjax.scipy.ndimage
. In particular, I thinkresize
is exactly what you're looking for, though I'm not sure JAX has 100% of the options found in PyTorch.