aboutsummaryrefslogtreecommitdiff
path: root/decoder.py
diff options
context:
space:
mode:
authorAnthony Wang2024-04-28 19:33:51 -0400
committerAnthony Wang2024-04-28 19:33:51 -0400
commit80b6b746b82be2046847a658711b6845c39bc52f (patch)
treed717335c0251b488613bdc0f43e4115ad1003b27 /decoder.py
parent5259da7b02e62f81c383fa67e06cd7ee554e6301 (diff)
Format code, update usage instructions
Diffstat (limited to 'decoder.py')
-rw-r--r--decoder.py4
1 files changed, 2 insertions, 2 deletions
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()