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

ajuste no capture raw

This commit is contained in:
Diego Freitas 2026-04-13 13:12:25 -03:00
parent ad81df8fa6
commit 5e5c9ea34c
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -96,7 +96,7 @@ def save_sample_raw4(
): ):
""" """
Salva: Salva:
- RAW4 como .npy float32 (4,H,W) - RAW4 como .raw float32 (4,H,W)
- preview RGB como .png - preview RGB como .png
- metadados como .json - metadados como .json
dentro de base_dir. dentro de base_dir.
@ -257,7 +257,7 @@ def main():
}, },
"note": "autosave", "note": "autosave",
} }
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 (auto): {os.path.basename(raw_path)}" last_msg = f"SALVO (auto): {os.path.basename(raw_path)}"
last_msg_t = now last_msg_t = now
@ -295,6 +295,10 @@ def main():
"exp_raw": int(exp_raw) if exp_raw is not None else None, "exp_raw": int(exp_raw) if exp_raw is not None else None,
"gain_a": int(gain_a) if gain_a is not None else None, "gain_a": int(gain_a) if gain_a is not None else None,
"gain_d": int(gain_d) if gain_d is not None else None, "gain_d": int(gain_d) if gain_d is not None else None,
"apply_ir_comp": apply_ir_comp,
"ir_k_r": ir_k_r,
"ir_k_g": ir_k_g,
"ir_k_b": ir_k_b,
"ae_dbg": { "ae_dbg": {
k2: (float(v2) if isinstance(v2, (int, float, np.floating)) else v2) k2: (float(v2) if isinstance(v2, (int, float, np.floating)) else v2)
for k2, v2 in ae_dbg.items() for k2, v2 in ae_dbg.items()