From c50dab64cbc216e14a6231d52665314422d592bd Mon Sep 17 00:00:00 2001 From: Diego Freitas Date: Thu, 16 Apr 2026 18:38:14 -0300 Subject: [PATCH] ajustes no script de captura em raw --- Python/OAK/datasets/_0_capture_raw.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/Python/OAK/datasets/_0_capture_raw.py b/Python/OAK/datasets/_0_capture_raw.py index 26c4f02bf..48dd8f236 100644 --- a/Python/OAK/datasets/_0_capture_raw.py +++ b/Python/OAK/datasets/_0_capture_raw.py @@ -186,6 +186,11 @@ def main(): cam.configure_fps(20) cam.start_streaming() + apply_ir_comp = True + ir_k_r = 0.8 + ir_k_g = 0.4 + ir_k_b = 0.9 + while True: t0 = time.time() raw4_base, dbg = cam.grab_raw4(out_h=RAW_H, out_w=RAW_W, timeout_ms=2000, do_ae=True) @@ -196,10 +201,6 @@ def main(): gain_a = dbg.get("gain_a", None) gain_d = dbg.get("gain_d", None) - apply_ir_comp = True - ir_k_r = 0.4 - ir_k_g = 0.1 - ir_k_b = 0.5 bgr = make_bgr_preview_from_raw(raw4_base, rgirb=True, preview_fast=upscale > 0, preview_scale=upscale, apply_ir_comp=apply_ir_comp, ir_k_r=ir_k_r, ir_k_g=ir_k_g, ir_k_b=ir_k_b) # FPS @@ -305,7 +306,7 @@ def main(): }, "note": "manual", } - rgb_clean_bgr_save = make_bgr_preview_from_raw(raw4_base, rgirb=True, preview_fast=False) + rgb_clean_bgr_save = make_bgr_preview_from_raw(raw4_base, rgirb=True, preview_fast=False, apply_ir_comp=apply_ir_comp, ir_k_r=ir_k_r, ir_k_g=ir_k_g, ir_k_b=ir_k_b) raw_path, _, _ = save_sample_raw4(session_dir, raw4_base, rgb_clean_bgr_save, meta) last_msg = f"SALVO (manual): {os.path.basename(raw_path)}" last_msg_t = time.time()