Readme update

haydn-jones · Mar 23, 2024 · df426c9 · df426c9
1 parent 3efe998
commit df426c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,6 +1,6 @@
 # DiffJPEG: A Jax Implementation
 
-This is a Jax implementation of the differentiable JPEG compression algorithm, based on the [PyTorch implementation](https://github.com/mlomnitz/DiffJPEG).
+This is a Jax implementation of the differentiable JPEG compression algorithm, based on the [PyTorch implementation](https://github.com/mlomnitz/DiffJPEG) and some of the modifications found in [this repository](https://github.com/necla-ml/Diff-JPEG) to improve quality at high compression rates.
 
 
 ## Requirements
@@ -17,7 +17,7 @@ pip install diffjpeg_jax
 
 ## Usage
 
-Unlike the PyTorch version, this is ML library agnostic, so it simply is implemented as a function.
+Unlike the PyTorch version, this is ML library agnostic, so it simply is implemented as a function. Inputs should be in the range `[0, 255]` and in the format `(H, W, C)`.
 
 ```python