From f3321b6782aff20df2c029239cc0f8b9a59ffde7 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Sat, 23 Jan 2021 00:16:58 +0800
Subject: Evaluation bug fixes and code review

1. Return full cached clip in evaluation
2. Add multi-iter checkpoints support in evaluation
3. Remove duplicated code while transforming
---
 utils/dataset.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'utils/dataset.py')

diff --git a/utils/dataset.py b/utils/dataset.py
index cd8b0f1..bbd42c3 100644
--- a/utils/dataset.py
+++ b/utils/dataset.py
@@ -201,6 +201,9 @@ class CASIAB(data.Dataset):
                 self._cached_clips[clip_name] = clip
             else:  # Load cache
                 cached_clip = self._cached_clips[clip_name]
+                # Return full clips while evaluating
+                if not self._is_train:
+                    return cached_clip
                 cached_clip_frame_names \
                     = self._cached_clips_frame_names[clip_path]
                 # Index the original clip via sampled frame names
-- 
cgit v1.2.3