1. Delete unused transform function

2. Reorganize the initialization cache dicts

1 files changed, 5 insertions, 6 deletions

diff --git a/utils/dataset.py b/utils/dataset.py
index 63cffa5..f41a9c5 100644
--- a/utils/dataset.py
+++ b/utils/dataset.py
@@ -14,10 +14,6 @@ ClipLabels = NewType('ClipLabels', Set[str])
 ClipConditions = NewType('ClipConditions', Set[str])
 ClipViews = NewType('ClipViews', Set[str])
 
-default_frame_transform = transforms.Compose([
-    transforms.Resize(size=(64, 32))
-])
-
 
 class CASIAB(data.Dataset):
     """CASIA-B multi-view gait dataset"""
@@ -89,6 +85,11 @@ class CASIAB(data.Dataset):
         #   set of three attributes above
         self.metadata = Dict[str, Set[str]]
 
+        # Dictionaries for indexing frames and frame names by clip name
+        # and chip path when cache is on
+        self._cached_clips_frame_names: Optional[Dict[str, List[str]]] = None
+        self._cached_clips: Optional[Dict[str, torch.Tensor]] = None
+
         clip_names = sorted(os.listdir(self.root_dir))
 
         if self.is_train:
@@ -166,8 +167,6 @@ class CASIAB(data.Dataset):
                 'views': set(self.views.tolist())
             }
 
-        self._cached_clips_frame_names: Optional[Dict[str, List[str]]] = None
-        self._cached_clips: Optional[Dict[str, torch.Tensor]] = None
         if self.cache_on:
             self._cached_clips_frame_names = dict()
             self._cached_clips = dict()