Skip to content

Fix/563 eval model dict csv #617

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
prawnsgupta wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Fix/563 eval model dict csv #617

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
234980a
Fix #562: pass image tensor to resize_image() in TF segmentation t_in
prawnsgupta Jul 13, 2026
3fa5abc
Fix #563: handle detection dict result when writing --out_fname CSV
prawnsgupta Jul 13, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion perceptionmetrics/cli/eval_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
import os

import click

from perceptionmetrics import cli
Expand Down Expand Up @@ -206,7 +208,12 @@ def eval_model(
predictions_outdir=predictions_outdir,
results_per_sample=predictions_outdir is not None,
)
results.to_csv(out_fname)

# Detection models return a dict with metrics_df and metrics_factory;
# other tasks return the metrics DataFrame directly.
metrics_df = results.get("metrics_df") if isinstance(results, dict) else results
os.makedirs(os.path.dirname(os.path.abspath(out_fname)), exist_ok=True)
metrics_df.to_csv(out_fname)

return results

Expand Down
1 change: 1 addition & 0 deletions perceptionmetrics/models/tf_segmentation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -273,6 +273,7 @@ def t_in(image):

if "resize" in self.model_cfg:
tensor = resize_image(
tensor,
method="bilinear",
width=self.model_cfg["resize"].get("width", None),
height=self.model_cfg["resize"].get("height", None),
Expand Down