evan/ewconfig
1
Fork 0
Code

Add a squish mode

Browse Source
This commit is contained in:
Evan Pratten 2024-04-26 10:56:13 -04:00
parent c7f9f952af
commit acc14b9555
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
scripts/leap-view
View File

@ -65,7 +65,10 @@ def main() -> int:
ap.add_argument( ap.add_argument(
"--upscale", "--upscale",
help="Upscaling factor", help="Upscaling factor",
type=int, type=float,
)
ap.add_argument(
"--squish", help="Downscale the image before upscaling", action="store_true"
) )
ap.add_argument( ap.add_argument(
"--average-frames", help="Number of frames to average", type=int, default=0 "--average-frames", help="Number of frames to average", type=int, default=0
@ -192,6 +195,14 @@ def main() -> int:
left_frame = normalize_brightness(left_frame, args.brightness_normalization) left_frame = normalize_brightness(left_frame, args.brightness_normalization)
right_frame = normalize_brightness(right_frame, args.brightness_normalization) right_frame = normalize_brightness(right_frame, args.brightness_normalization)
# Average down by a 2x2 square
if args.squish:
left_frame = cv2.resize(left_frame, (width // 2, height // 2))
right_frame = cv2.resize(right_frame, (width // 2, height // 2))
left_frame = cv2.resize(left_frame, (width, height))
right_frame = cv2.resize(right_frame, (width, height))
# If we should be recording the video # If we should be recording the video
if args.record: if args.record:
# Create a new frame that is twice as wide with both images side by side # Create a new frame that is twice as wide with both images side by side
@ -207,10 +218,10 @@ def main() -> int:
# If we need to do upscaling, do it now # If we need to do upscaling, do it now
if args.upscale: if args.upscale:
left_frame = cv2.resize( left_frame = cv2.resize(
left_frame, (width * args.upscale, height * args.upscale) left_frame, (int(width * args.upscale), int(height * args.upscale))
) )
right_frame = cv2.resize( right_frame = cv2.resize(
right_frame, (width * args.upscale, height * args.upscale) right_frame, (int(width * args.upscale), int(height * args.upscale))
) )
# Show the frame # Show the frame