From 57a2ecb485d3814b85f4c22c9b48f1de053750eb Mon Sep 17 00:00:00 2001 From: Anthony Wang Date: Mon, 13 May 2024 19:45:14 -0400 Subject: Revert "Use medians for deciding color, seems to work well in practice" This reverts commit 05e4dfbd5b9ddee59f6d42a608612f11102b61fe. --- decoder.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/decoder.py b/decoder.py index 235a5f6..b23d623 100644 --- a/decoder.py +++ b/decoder.py @@ -99,11 +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]) - rmed = np.median(frame[:,:,0]) - gmed = np.median(frame[:,:,1]) - bmed = np.median(frame[:,:,2]) - frame = np.dstack((frame[:,:,0] > rmed, frame[:,:,1] > gmed, frame[:,:,2] > bmed)).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