Fix conflicts in start idx.

open-mmlab · Jul 28, 2020 · b3f6166 · b3f6166
1 parent b837bcb
commit b3f6166
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mmaction/datasets/pipelines/loading.py b/mmaction/datasets/pipelines/loading.py
@@ -403,9 +403,9 @@ def _get_proposal_clips(self, proposal, num_frames):
         valid_ending = min(num_frames - ori_clip_len + 1,
                            end_frame + int(duration * self.aug_ratio[1]))
 
-        valid_starting_length = (
-            start_frame - valid_starting - ori_clip_len + 1)
-        valid_ending_length = valid_ending - end_frame - ori_clip_len + 1
+        valid_starting_length = (start_frame - valid_starting +
+                                 1) - ori_clip_len
+        valid_ending_length = (valid_ending - end_frame + 1) - ori_clip_len
 
         if self.mode == 'train':
             starting_offsets = self._get_train_indices(valid_starting_length,
@@ -464,7 +464,7 @@ def _get_test_clips(self, num_frames):
         """
         ori_clip_len = self.clip_len * self.frame_interval
         return np.arange(
-            0, num_frames - ori_clip_len, self.test_interval, dtype=int) + 1
+            0, num_frames - ori_clip_len, self.test_interval, dtype=int)
 
     def _sample_clips(self, num_frames, proposals):
         """Choose clip offsets for the video in a given mode.