Zendion/agrobot_base
Zendion
/
agrobot_base
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ajustes no script de captura em raw

This commit is contained in:
Diego Freitas 2026-04-16 18:38:14 -03:00
parent 5e5c9ea34c
commit c50dab64cb
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Python/OAK/datasets/_0_capture_raw.py
View File

@ -186,6 +186,11 @@ def main():
cam.configure_fps(20) cam.configure_fps(20)
cam.start_streaming() cam.start_streaming()
apply_ir_comp = True
ir_k_r = 0.8
ir_k_g = 0.4
ir_k_b = 0.9
while True: while True:
t0 = time.time() t0 = time.time()
raw4_base, dbg = cam.grab_raw4(out_h=RAW_H, out_w=RAW_W, timeout_ms=2000, do_ae=True) 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_a = dbg.get("gain_a", None)
gain_d = dbg.get("gain_d", 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) 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 # FPS
@ -305,7 +306,7 @@ def main():
}, },
"note": "manual", "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) 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 = f"SALVO (manual): {os.path.basename(raw_path)}"
last_msg_t = time.time() last_msg_t = time.time()