Skip to content
This repository was archived by the owner on Feb 22, 2020. It is now read-only.

Commit 1777967

Browse files
committed
fix(encoder):
fix mixture encoder
1 parent 95f03c5 commit 1777967

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gnes/encoder/video/incep_mixture.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,8 @@ def post_init(self):
106106
def encode(self, videos: List['np.ndarray'], *args, **kwargs) -> np.ndarray:
107107
ret = []
108108
v_len = [len(v) for v in videos]
109-
pos_start = [0] + [sum(v_len[:i]) for i in range(1, len(v_len)-1)]
110-
pos_end = [sum(v_len[:i]) for i in range(len(v_len))]
109+
pos_start = [0] + [sum(v_len[:i+1]) for i in range(len(v_len)-1)]
110+
pos_end = [sum(v_len[:i+1]) for i in range(len(v_len))]
111111
max_len = min(max(v_len), self.max_frames)
112112

113113
img = [im for v in videos for im in v]

0 commit comments

Comments
 (0)