diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 00:41:30 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 00:41:30 +0800 |
| commit | f8fe82326479d4ea62d7da3dd8f25f1c09cf11bc (patch) | |
| tree | b8cd0414a86eb8be8660e76bfca4764b5766ce59 /utils/dataset.py | |
| parent | d750dd9dafe3cda3b1331ad2bfecb53c8c2b1267 (diff) | |
| parent | f3321b6782aff20df2c029239cc0f8b9a59ffde7 (diff) | |
Merge branch 'master' into python3.8
Diffstat (limited to 'utils/dataset.py')
| -rw-r--r-- | utils/dataset.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/utils/dataset.py b/utils/dataset.py index ea678fb..6b2a991 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 |