multiple cameras and frame saving in video.py

This commit is contained in:
Alexander Mordvintsev 2011-06-10 07:54:21 +00:00
parent fac611337f
commit 787fe6a93f
2 changed files with 23 additions and 6 deletions

View File

@ -16,7 +16,7 @@ if len(sys.argv) > 1:
img = cv2.imread(fn)
else:
sz = 4096
print 'generating %dx%d procudural image ...' % (sz, sz)
print 'generating %dx%d procedural image ...' % (sz, sz)
img = np.zeros((sz, sz), np.uint8)
track = np.cumsum(np.random.rand(500000, 2)-0.5, axis=0)
track = np.int32(track*10 + (sz/2, sz/2))

View File

@ -49,13 +49,30 @@ def create_capture(source):
if __name__ == '__main__':
import sys
try: fn = sys.argv[1]
except: fn = 'synth:bg=../cpp/lena.jpg:noise=0.1'
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('sources', nargs='*', default=['synth:bg=../cpp/lena.jpg:noise=0.1'])
parser.add_argument('-shotdir', nargs=1, default='.')
args = parser.parse_args()
print args
cap = create_capture(fn)
print 'Press SPACE to save current frame'
caps = map(create_capture, args.sources)
shot_idx = 0
while True:
ret, img = cap.read()
cv2.imshow('img', img)
imgs = []
for i, cap in enumerate(caps):
ret, img = cap.read()
imgs.append(img)
cv2.imshow('capture %d' % i, img)
ch = cv2.waitKey(1)
if ch == 27:
break
if ch == ord(' '):
for i, img in enumerate(imgs):
fn = '%s/shot_%d_%03d.bmp' % (args.shotdir[0], i, shot_idx)
cv2.imwrite(fn, img)
print fn, 'saved'
shot_idx += 1