From 80b6b746b82be2046847a658711b6845c39bc52f Mon Sep 17 00:00:00 2001 From: Anthony Wang Date: Sun, 28 Apr 2024 19:33:51 -0400 Subject: Format code, update usage instructions --- README.md | 2 +- decoder.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 776d41c..9857d2a 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ 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` +Copy the flags printed by the encoder and pass them to the decoder: `python decoder.py -o out FLAGS` Formatting: `black -l 120 *.py` diff --git a/decoder.py b/decoder.py index 2738939..d541d6e 100644 --- a/decoder.py +++ b/decoder.py @@ -60,7 +60,7 @@ while data is None: # raw_frame is a uint8 BE CAREFUL if type(args.input) == int: X, Y = raw_frame.shape[:2] - raw_frame = raw_frame[X // 4:3 * X // 4, Y // 4: 3 * Y // 4] + raw_frame = raw_frame[X // 4 : 3 * X // 4, Y // 4 : 3 * Y // 4] cv2.imshow("", raw_frame) cv2.waitKey(1) raw_frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB) @@ -127,7 +127,7 @@ while data is None: ) ) # erase_pos = bytearray(np.where(frame == 0)[0]) if args.erasure else bytearray() - data = decoder.decode(bytes(rsc.decode(bytearray(np.packbits(frame) ^ frame_xor))[0][:args.psize])) + data = decoder.decode(bytes(rsc.decode(bytearray(np.packbits(frame) ^ frame_xor))[0][: args.psize])) print("Decoded frame") except KeyboardInterrupt: sys.exit() -- cgit v1.2.3-70-g09d2