When HeatMapAnnotator is called on a fresh annotator with empty detections (common on the first frames of a video before the model produces any output), self.heat_mask is all zeros, so temp / temp.max() raises RuntimeWarning: invalid value encountered in divide and produces nan/inf in-flight. Skip the normalisation when temp.max() == 0; the resulting all-zero heat mask filters out via the > 0 check below, so the scene is returned unchanged. - Fix `kernel_size: int = 25` → `int | None = 25`; document None disables blur - Add Note to annotate docstring: empty detections returns scene unchanged - Add happy path test: single detection must produce visible heat output - Add stateful tests: empty→real and real→empty sequence coverage --------- Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: Claude Code <noreply@anthropic.com> |
||
|---|---|---|
| .. | ||
| annotators | ||
| assets | ||
| classification | ||
| dataset | ||
| detection | ||
| draw | ||
| geometry | ||
| key_points | ||
| metrics | ||
| tracker | ||
| utils | ||
| __init__.py | ||
| conftest.py | ||
| helpers.py | ||