dm_pix

dm_pix copied to clipboard

Add elastic transformation.

copybara-service[bot]

Gaussian blur cpu performance

12

I have been doing some experiments with PIX since it allows computing image augmentations in the GPU in contrast to torchvision which computes in the CPU and requires multiple workers...

mgoulao

About the requirement of jax version.

1

I noticed that all versions of dm-pix library require `jax>=0.4.16`, even the `0.1.0` version. Is that necessary? I wonder if this can be modified so that I can use dm-pix...

AlbertHuyb

random_flip_left_right does not support HWC

1

https://dm-pix.readthedocs.io/en/latest/api.html#random-flip-left-right Despite being documented to say so, there is no way to actually pass in the channel_axis to this function. So you cannot set the image format to be HWC....

Logon27