emdee/toxygen
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fixed bug with video sending

Browse source
This commit is contained in:
ingvar1995 2017-06-13 21:50:00 +03:00
parent 6d705deb55
commit d1e90c6aef
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
toxygen/calls.py
View file

@ -292,17 +292,17 @@ class AV:
""" """
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2YUV_I420) frame = cv2.cvtColor(frame, cv2.COLOR_BGR2YUV_I420)
y = frame[:self._video_height, :].tolist() y = frame[:self._video_height, :]
y = list(itertools.chain.from_iterable(y)) y = list(itertools.chain.from_iterable(y))
u = np.zeros((self._video_width // 2, self._video_height // 2), dtype=np.int) u = np.zeros((self._video_height // 2, self._video_width // 2), dtype=np.int)
u[::2, :] = frame[self._video_height:self._video_height * 5 // 4, :self._video_height // 2] u[::2, :] = frame[self._video_height:self._video_height * 5 // 4, :self._video_width // 2]
u[1::2, :] = frame[self._video_height:self._video_height * 5 // 4, self._video_height // 2:] u[1::2, :] = frame[self._video_height:self._video_height * 5 // 4, self._video_width // 2:]
u = list(itertools.chain.from_iterable(u)) u = list(itertools.chain.from_iterable(u))
v = np.zeros((self._video_width // 2, self._video_height // 2), dtype=np.int) v = np.zeros((self._video_height // 2, self._video_width // 2), dtype=np.int)
v[::2, :] = frame[self._video_height * 5 // 4:, :self._video_height // 2] v[::2, :] = frame[self._video_height * 5 // 4:, :self._video_width // 2]
v[1::2, :] = frame[self._video_height * 5 // 4:, self._video_height // 2:] v[1::2, :] = frame[self._video_height * 5 // 4:, self._video_width // 2:]
v = list(itertools.chain.from_iterable(v)) v = list(itertools.chain.from_iterable(v))
return bytes(y), bytes(u), bytes(v) return bytes(y), bytes(u), bytes(v)