import Image
from PyQt4 import QtGui
from PyQt4 import QtCore, QtGui
from CLWindow import CLWindow
from CLCanvas import CLCanvas
from Colorize import Colorize
from Brush import Brush
from GrowCut import GrowCut
import sys
img = Image.open("/Users/marcdeklerk/msc/code/dataset/processed/source/800x600/GT04.png").convert('RGBA')
tri = Image.open("/Users/marcdeklerk/msc/code/dataset/processed/trimap1/800x600/GT04.png").convert('RGBA')
app = QtGui.QApplication(sys.argv)
canvas = CLCanvas(img.size)
window = CLWindow(canvas)
context = canvas.context
devices = context.get_info(cl.context_info.DEVICES)
queue = cl.CommandQueue(context, properties=cl.command_queue_properties.PROFILING_ENABLE)
shape = (img.size[1], img.size[0])
shape = roundUp(shape, SharedMatting.lw)
dim = (shape[1], shape[0])
hImg = padArray2D(np.array(img).view(np.uint32).squeeze(), shape, 'edge')
dImgGC = cl.Image(context,
cl.mem_flags.READ_ONLY,
cl.ImageFormat(cl.channel_order.RGBA, cl.channel_type.UNSIGNED_INT8),
dim
from Colorize import Colorize
from clutil import roundUp, padArray2D
from QuickBrush import QuickBrush
from Buffer2D import Buffer2D
from Image2D import Image2D
import numpy as np
import pyopencl as cl
cm = cl.mem_flags
img = Image.open("/Users/marcdeklerk/msc/code/dataset/processed/source/800x600/GT04.png")
if img.mode != 'RGBA':
img = img.convert('RGBA')
app = QtGui.QApplication(sys.argv)
canvas = CLCanvas(img.size)
window = CLWindow(canvas)
clContext = canvas.context
devices = clContext.get_info(cl.context_info.DEVICES)
queue = cl.CommandQueue(clContext, properties=cl.command_queue_properties.PROFILING_ENABLE)
shapeNP = (img.size[1], img.size[0])
shapeNP = roundUp(shapeNP, QuickBrush.lWorksize)
shapeCL = (shapeNP[1], shapeNP[0])
hImg = padArray2D(np.array(img).view(np.uint32).squeeze(), shapeNP, 'edge')
dImg = Image2D(clContext,
cl.mem_flags.READ_ONLY,
cl.ImageFormat(cl.channel_order.RGBA, cl.channel_type.UNSIGNED_INT8),
shapeCL
img = Image.open("/Users/marcdeklerk/msc/code/dataset/processed/source/800x600/GT04.png")
if img.mode != 'RGBA':
img = img.convert('RGBA')
shape = (img.size[1], img.size[0])
hImg = padArray2D(np.array(img).view(np.uint32).squeeze(), roundUp(shape, GraphCut.lWorksize), 'edge')
width = hImg.shape[1]
height = hImg.shape[0]
dim = (width, height)
shape = (height, width)
app = QtGui.QApplication(sys.argv)
canvas = CLCanvas(dim)
window = CLWindow(canvas)
context = canvas.context
queue = cl.CommandQueue(context)
dImg = Buffer2D(context, cm.READ_WRITE | cm.COPY_HOST_PTR, hostbuf=hImg)
devices = context.get_info(cl.context_info.DEVICES)
gc = GraphCut(context, devices, dImg)
window.resize(1000, 700)
hSrc = np.load('scoreFg.npy').reshape(shape)
hSink = np.load('scoreBg.npy').reshape(shape)
