diff options
| -rw-r--r-- | decoder.py | 8 |
1 files changed, 2 insertions, 6 deletions
@@ -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( ( |