generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

video synth uses cv2.randn for noise -- much faster than np.random.normal

Browse Source
This commit is contained in:
Alexander Mordvintsev
2011-06-09 08:21:37 +00:00
parent 2e7e6ae84b
commit 07a4e52093
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
samples/python2/video.py
View File

@@ -20,11 +20,13 @@ class VideoSynth(object):
def read(self, dst=None): def read(self, dst=None):
w, h = self.frame_size w, h = self.frame_size
buf = np.zeros((h, w, 3), np.uint8) if self.bg is None:
if self.bg is not None: buf = np.zeros((h, w, 3), np.uint8)
buf[:] = self.bg else:
buf = self.bg.copy()
if self.noise > 0.0: if self.noise > 0.0:
noise = np.random.normal(scale = 255*self.noise, size=(h, w, 3)) noise = np.zeros((h, w, 3), np.int8)
cv2.randn(noise, np.zeros(3), np.ones(3)*255*self.noise)
buf = cv2.add(buf, noise, dtype=cv2.CV_8UC3) buf = cv2.add(buf, noise, dtype=cv2.CV_8UC3)
return True, buf return True, buf