You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
804 B
26 lines
804 B
import cv2 as cv |
|
import argparse |
|
|
|
|
|
def main(): |
|
argparser = argparse.ArgumentParser(description='Vizualization of the SyntheticSequenceGenerator.') |
|
|
|
argparser.add_argument('-b', '--background', help='Background image.', required=True) |
|
argparser.add_argument('-o', '--obj', help='Object image. It must be strictly smaller than background.', required=True) |
|
args = argparser.parse_args() |
|
|
|
bg = cv.imread(args.background) |
|
obj = cv.imread(args.obj) |
|
generator = cv.bgsegm.createSyntheticSequenceGenerator(bg, obj) |
|
|
|
while True: |
|
frame, mask = generator.getNextFrame() |
|
cv.imshow('Generated frame', frame) |
|
cv.imshow('Generated mask', mask) |
|
k = cv.waitKey(int(1000.0 / 30)) |
|
if k == 27: |
|
break |
|
|
|
|
|
if __name__ == '__main__': |
|
main()
|
|
|