Ejemplo n.º 1
0
from inception.image.matte.bordermatte import *

if __name__ == '__main__':
    from inception.image.image import Image
    #image = Image.from_filepath("../../../../test/images/buff_small.png")#traditional-buffets-and-sideboards.jpg")
    image = Image.from_filepath("../../../../test/images/traditional-buffets-and-sideboards.jpg")
    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
    """
    
Ejemplo n.º 2
0
from inception.image.shadow.shadow import *

if __name__ == '__main__':
    import os
    from inception.image.image import Image
    from inception.image.matte.bordermatte import alphamatte
    from inception.image.operation.merge import MergeOperation
    from inception.image.operation.scale import ScaleOperation
    from PIL import Image as PILImage

    image = Image.from_filepath("../../../../test/images/traditional-buffets-and-sideboards.jpg")
    bg = Image.from_filepath("../../../../test/images/vanishing.jpg")
    
    result = alphamatte(image.data)
    
    filename = image.filename
    image.to_rgba()
    image[..., 3] = result
    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)
Ejemplo n.º 3
0
from inception.image.analyze import *

if __name__ == '__main__':
    import os
    import urllib, cStringIO
    from inception.image.image import Image
    urlfilepath = "../../../test/urls.txt"
    urls = []
    if os.path.exists(urlfilepath):
        with open(urlfilepath) as fileobj:
            urls = [l.strip() for l in fileobj.readlines() if l.strip()]
    
    for url in urls:
        fileobj = cStringIO.StringIO(urllib.urlopen(url).read())
        img = Image.from_any(fileobj)
        print validate_as_foreground(img), url
        
        
Ejemplo n.º 4
0
 print("DONE")
 """
 from inception.image.image import Image
 from inception.image.operation import merge
 import os, glob
 testdir = os.path.abspath(os.path.dirname(__file__) + '/../../../../test/images/statadjust')
 #mattes = glob.glob(testdir + "/brigh*_small_matte.jpg")
 mattes = glob.glob(testdir + "/ice_stand_small_matte.jpg")
 #mattes.extend(glob.glob(testdir + "/col*_small_matte.jpg"))
 mattes.extend(glob.glob(testdir + "/dance_small_matte.jpg"))
 backgrounds = [a.rsplit("_matte.jpg",1)[0] + ".jpg" for a in mattes]
 
 foregroundImages = []
 backgroundImages = []
 for i, matte in enumerate(mattes):
     backgroundImages.append(Image.from_filepath(backgrounds[i]))
     matteImage = Image.from_filepath(matte)
     foregroundImage = backgroundImages[-1].clone()
     foregroundImage.to_rgba()
     foregroundImage[..., 3] = linear_to_srgb(getluminance(srgb_to_linear(matteImage.data)))
     foregroundImages.append(foregroundImage)
     foregroundImage.filename = matteImage.filename
 
 # test the images
 for i in range(len(backgroundImages)):
     bg = backgroundImages[i]
     fg = i + 1 if (i < len(backgroundImages) - 1) else 0
     fg = foregroundImages[fg]
     
     print("-------- Running on %s over %s ----------" % (os.path.basename(fg.filename), os.path.basename(bg.filename)))
     results = statadjust(fg.data, bg.data, intermediary_results=True)