args = parser.parse_args()
logging.info('Loading original image %s' % (args.original_image))
org_img = load_image(args.original_image)
logging.info('Method to create mask')
mask_type = args.mask
logging.info('Loading style...')
sty_list = args.style
masks = args.mask
mask_list = []
for i in masks:
mask_list.append(load_image(i))
# make a transferor with all inputs, get transfered img.
transferor = Transfer.Transferor()
transferor.get_original(org_img)
transferor.get_styles(sty_list)
transferor.transfer_images()
transfered_imgs = transferor.return_transfered_list()
# make a blendor with all input, get blended img.
blendor = Blend.Blendor()
blendor.get_input(org_img, transfered_imgs)
blendor.get_mask(mask_list)
blendor.blend()
blended_img = blendor.return_output()
cv2.imwrite(args.transfered_image, blended_img)
