Add usage instructions to README.md

2 files changed, 7 insertions, 4 deletions

diff --git a/README.md b/README.md
index c2ac901..98eb594 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,11 @@
 # 6.8301 Project
 
+Encode: `python encoder.py -i in -o vid.mkv`
+
+Play video: `mpv --scale=nearest --fullscreen --loop --no-keepaspect vid.mkv`
+
+Decode: `python decoder.py -o out`
+
 Formatting: `black -l 120 *.py`
 
 TODO:
diff --git a/encoder.py b/encoder.py
index 15ee859..f61ca9c 100644
--- a/encoder.py
+++ b/encoder.py
@@ -67,10 +67,7 @@ def get_frame():
             ),
         )
     )
-    return np.stack(
-        ((frame & 0b0001) * 255, (frame >> 1 & 0b0011) * 255 / 3, (frame >> 3) * 255),
-        axis=-1,
-    ).astype(np.uint8)
+    return np.stack(((frame & 0b0001) * 255, (frame >> 1 & 0b0011) * 85, (frame >> 3) * 255), axis=-1).astype(np.uint8)
 
 
 if args.mix: