From f0c83727657cd54b97f47f36d4d133938df54410 Mon Sep 17 00:00:00 2001 From: cameron-a-johnson <43187095+cameron-a-johnson@users.noreply.github.com> Date: Thu, 21 Nov 2024 13:04:45 -0500 Subject: [PATCH] Update tcn_hpl/callbacks/plot_metrics.py Co-authored-by: Paul Tunison <735270+Purg@users.noreply.github.com> --- tcn_hpl/callbacks/plot_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tcn_hpl/callbacks/plot_metrics.py b/tcn_hpl/callbacks/plot_metrics.py index f5c3504d4..a78d842d0 100644 --- a/tcn_hpl/callbacks/plot_metrics.py +++ b/tcn_hpl/callbacks/plot_metrics.py @@ -371,8 +371,8 @@ def on_test_epoch_end( acts_dset.fpath = self._preds_dset_output_fpath acts_dset.dataset['videos'] = truth_dset.dataset['videos'] acts_dset.dataset['images'] = truth_dset.dataset['images'] - acts_dset.index.build(acts_dset) acts_dset.dataset['categories'] = truth_dset.dataset['categories'] + acts_dset.index.build(acts_dset) # Create numpy lookup tables for i in range(len(all_preds)): frame_index = all_source_frames[i].item()