Add top-row picture-in-picture guard insets

This commit is contained in:
Meta Vens 2026-04-22 12:46:02 +01:00
commit 6e5eeac6bb
2 changed files with 29 additions and 0 deletions

Binary file not shown.

Before

Width:  |  Height:  |  Size: 383 KiB

After

Width:  |  Height:  |  Size: 390 KiB

Before After
Before After

View file

@ -340,6 +340,34 @@ def main() -> None:
target_width=ai_result.inference.target_width,
target_height=ai_result.inference.target_height,
)
vector_guard_bbox = _cell_bbox_to_source_bbox(
tuple(args.guard_cell_bbox),
source_width=vector_source.shape[1],
source_height=vector_source.shape[0],
target_width=args.vector_target_size,
target_height=args.vector_target_size,
)
vector_lanczos_preview = nearest_resize(vector_lanczos, width=vector_source.shape[1], height=vector_source.shape[0])
vector_repixelized_preview = nearest_resize(
vector_result.output_rgba,
width=vector_source.shape[1],
height=vector_source.shape[0],
)
for panel in vector_panels:
_draw_source_bbox(
panel,
vector_guard_bbox,
source_width=vector_source.shape[1],
source_height=vector_source.shape[0],
)
vector_crops = [
_crop_rgba(vector_source, vector_guard_bbox),
_crop_rgba(vector_lanczos_preview, vector_guard_bbox),
_crop_rgba(vector_repixelized_preview, vector_guard_bbox),
]
for panel, crop in zip(vector_panels, vector_crops, strict=True):
_add_pip_inset(panel, _build_pip_inset(crop))
ai_lanczos_preview = nearest_resize(ai_lanczos, width=ai_source.shape[1], height=ai_source.shape[0])
ai_repixelized_preview = nearest_resize(ai_result.output_rgba, width=ai_source.shape[1], height=ai_source.shape[0])
for panel in ai_panels:
@ -395,6 +423,7 @@ def main() -> None:
"ai_inferred_width": ai_result.inference.target_width,
"ai_inferred_height": ai_result.inference.target_height,
"guard_cell_bbox": list(args.guard_cell_bbox),
"vector_guard_source_bbox": list(vector_guard_bbox),
"guard_source_bbox": list(guard_bbox),
"out_sheet": str(out_sheet),
"out_guard_crop": str(out_guard_crop),