Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# PIX

<img src="https://raw.githubusercontent.com/google-deepmind/dm_pix/master/dm_pix/images/pix_logo.png" title="PIX logo" height=200 width=200>
<img src="https://raw.githubusercontent.com/google-deepmind/dm_pix/master/dm_pix/images/pix_logo.png" title="PIX logo" width=350>

PIX is an image processing library in [JAX], for [JAX].

Expand Down Expand Up @@ -39,11 +39,11 @@ $ pip install dm-pix

To use `PIX`, you just need to `import dm_pix as pix` and use it right away!

For example, let's assume to have loaded the JAX logo (available in
`examples/assets/jax_logo.jpg`) in a variable called `image` and we want to flip
For example, let's assume to have loaded the PIX logo (available in
`examples/assets/pix_logo.jpg`) in a variable called `image` and we want to flip
it left to right.

![JAX logo]
![PIX logo]

All it's needed is the following code!

Expand All @@ -58,7 +58,7 @@ flip_left_right_image = pix.flip_left_right(image)

And here is the result!

![JAX logo left-right]
![PIX logo left-right]

All the functions in PIX can be [`jax.jit`][jit]ed, [`jax.vmap`][vmap]ed and
[`jax.pmap`][pmap]ed, so all the following functions can take advantage of
Expand Down Expand Up @@ -145,6 +145,6 @@ Please read our [contributing guidelines](./CONTRIBUTING.md) and send us PRs!
[vmap]: https://jax.readthedocs.io/en/latest/jax.html#jax.vmap "jax.vmap documentation"

[`examples/`]: ./examples/
[JAX logo]: ./examples/assets/jax_logo.jpg
[JAX logo left-right]: ./examples/assets/flip_left_right_jax_logo.jpg
[PIX logo]: https://raw.githubusercontent.com/google-deepmind/dm_pix/master/examples/assets/pix_logo.jpg
[PIX logo left-right]: https://raw.githubusercontent.com/google-deepmind/dm_pix/master/examples/assets/flip_left_right_pix_logo.jpg
[`pyproject.toml`]: ./pyproject.toml
Binary file modified dm_pix/images/pix_logo.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file removed examples/assets/adjust_brightness_jax_logo.jpg
Binary file not shown.
Binary file removed examples/assets/adjust_contrast_jax_logo.jpg
Binary file not shown.
Binary file removed examples/assets/adjust_gamma_jax_logo.jpg
Binary file not shown.
Binary file removed examples/assets/adjust_hue_jax_logo.jpg
Binary file not shown.
Binary file removed examples/assets/flip_left_right_jax_logo.jpg
Binary file not shown.
Binary file added examples/assets/flip_left_right_pix_logo.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file removed examples/assets/flip_up_down_jax_logo.jpg
Binary file not shown.
Binary file removed examples/assets/gaussian_blur_jax_logo.jpg
Binary file not shown.
Binary file removed examples/assets/jax_logo.jpg
Binary file not shown.
Binary file added examples/assets/pix_logo.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
166 changes: 88 additions & 78 deletions examples/image_augmentation.ipynb

Large diffs are not rendered by default.

2 changes: 1 addition & 1 deletion examples/image_augmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ def main(_) -> None:


def _get_image():
return jnp.array(pil.open("./assets/jax_logo.jpg"), dtype=jnp.float32) / 255.
return jnp.array(pil.open("./assets/pix_logo.jpg"), dtype=jnp.float32) / 255.


def _imshow(image: jnp.ndarray) -> None:
Expand Down
Loading