fix(data): clip_len为多帧时下标超过图像数目

ZJCV · Oct 29, 2020 · daf01a8 · daf01a8
1 parent 61e3035
commit daf01a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tsn/data/datasets/clipsample/seg_sample.py b/tsn/data/datasets/clipsample/seg_sample.py
@@ -48,7 +48,8 @@ def _get_train_clips(self, num_frames):
             np.ndarray: Sampled frame indices in train mode.
         """
         one_clip_len = self.clip_len * self.frame_interval
-        avg_interval = (num_frames - one_clip_len + 1) // self.num_clips
+        # avg_interval = (num_frames - one_clip_len + 1) // self.num_clips
+        avg_interval = (num_frames - one_clip_len) // self.num_clips
 
         if avg_interval > 0:
             base_offsets = np.arange(self.num_clips) * avg_interval