image = Image.from_filepath("../../../../test/images/cookiejar.jpg") #image = Image.from_filepath("../../../../test/images/buff_small.png") import time t1 = time.time() print("Starting") result = alphamatte(image.data) t2 = time.time() print("Done (in %2.2f seconds)!" % (t2-t1)) filename = image.filename image.to_rgba() image[..., 3] = result """ image = image[...,0] image = Image(image) image.to_rgb() image[..., 0] = result image[..., 1] = result image[..., 2] = result """ from PIL import Image as PILImage green = Image.from_any(PILImage.new('RGBA', (image.shape[1], image.shape[0]), color=(0,255,0,255))) from inception.image.operation.merge import MergeOperation mop = MergeOperation([green, image]).run() mop.save(filename.rsplit('.',1)[0] + '_greencomp.png') Image.from_any(result).save(filename.rsplit('.',1)[0] + '_matte.png')
image = ScaleOperation(image, image.width/3.0).run() offset = 20, 200 pickle_path = bg.filename.rsplit('.', 1)[0] + '.scene.pkl' #pickle_path = bg.filename.rsplit('/',1)[0] + '/definitive.vanishing.pkl' import cPickle if (os.path.exists(pickle_path)): with open(pickle_path, 'r') as f: sd = cPickle.load(f) print("Detected vanishing points (x,y,z): %s" % sd.get_vanishing_points()) print("Focal length: %s" % sd.focal_length) print(sd.vpestimator.vanishing_points) else: sd = estimate_scene_description(bg) with open(pickle_path, 'w') as f: cPickle.dump(sd, f) bg.scene_description = sd shadow = Image.from_any(create_shadow(image, bg, offset=offset, scene_description=None, skip_soften=True)) image[...,3] *= .5 mop = MergeOperation([bg, shadow, image], offsets=[(0,0),(0,0),offset]).run() mop.save(filename.rsplit('.',1)[0] + '_shadowcomp.png') """ green = Image.from_any(PILImage.new('RGBA', (image.shape[1], image.shape[0]), color=(0,255,0,255))) mop = MergeOperation([green, shadow]).run() mop.save(filename.rsplit('.',1)[0] + '_greencomp.png') """ Image.from_any(shadow).save(filename.rsplit('.',1)[0] + '_matte.png')