aboutsummaryrefslogtreecommitdiff
path: root/decoder.py
diff options
context:
space:
mode:
Diffstat (limited to 'decoder.py')
-rw-r--r--decoder.py6
1 files changed, 1 insertions, 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(
(