diff options
Diffstat (limited to 'decoder.py')
| -rw-r--r-- | decoder.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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() |