diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 00:47:46 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 00:47:46 +0800 |
| commit | afe615408c4003a513811d900fe3edd119a735a5 (patch) | |
| tree | 7ab780ab400d3a768a52ab1446884d05f04e8119 /utils/dataset.py | |
| parent | c0c8299354bd41bfd668ff0fb3edb5997f590c5d (diff) | |
| parent | a040400d7caa267d4bfbe8e5520568806f92b3d4 (diff) | |
Merge branch 'python3.8' into python3.7
# Conflicts:
# models/model.py
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 |