From a1d77a466dcc694f0101d2588529ada8c858ed94 Mon Sep 17 00:00:00 2001 From: Anthony Wang Date: Sun, 12 May 2024 23:39:18 -0400 Subject: Minor tweaks for 0.3mbps -> 1.8 mbps --- decoder.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/decoder.py b/decoder.py index a1c0d51..b23d623 100644 --- a/decoder.py +++ b/decoder.py @@ -61,17 +61,13 @@ while data is None: print("Skipped") continue # raw_frame is a uint8 BE CAREFUL - if type(args.input) == int: - # Crop image to reduce camera distortion - X, Y = raw_frame.shape[:2] - 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) # Find positions and colors of corners X, Y = raw_frame.shape[:2] - cx, cy = X // 3, Y // 3 + cx, cy = X // 4, Y // 4 widx, wcol = find_corner(raw_frame[:cx, :cy], lambda B: np.sum(B, axis=2) - 2 * np.std(B, axis=2)) ridx, rcol = find_corner(raw_frame[:cx, Y - cy :], lambda B: B[:, :, 0] - B[:, :, 1] - B[:, :, 2]) ridx[1] += Y - cy @@ -103,7 +99,7 @@ while data is None: ) frame = cv2.warpPerspective(raw_frame, M, (args.width, args.height)) # Convert to new color space - frame = (np.squeeze(F @ (frame - origin)[..., np.newaxis]) >= 192).astype(np.uint8) + frame = (np.squeeze(F @ (frame - origin)[..., np.newaxis]) >= 160).astype(np.uint8) frame = np.packbits( np.concatenate( ( -- cgit v1.2.3-70-g09d2