[opengm] 31/40: read python demos

[Ghislain Vaillant]

This is an automated email from the git hooks/post-receive script.

ghisvail-guest pushed a commit to branch master
in repository opengm.

commit 2fa0a3c9bbb1e2639c2e558cc96094cd266b360f
Author: joergkappes <kappes at math.uni-heidelberg.de>
Date:   Thu Dec 10 20:00:37 2015 +0100

    read python demos
---
 src/tutorials/python/demo/demo1.py | 56 +++++++++++++++++++++++
 src/tutorials/python/demo/demo2.py | 56 +++++++++++++++++++++++
 src/tutorials/python/demo/demo3.py | 50 +++++++++++++++++++++
 src/tutorials/python/demo/demo4.py | 62 +++++++++++++++++++++++++
 src/tutorials/python/demo/demo5.py | 92 ++++++++++++++++++++++++++++++++++++++
 5 files changed, 316 insertions(+)

diff --git a/src/tutorials/python/demo/demo1.py b/src/tutorials/python/demo/demo1.py
new file mode 100644
index 0000000..a1661af
--- /dev/null
+++ b/src/tutorials/python/demo/demo1.py
@@ -0,0 +1,56 @@
+import opengm
+import vigra
+import numpy
+import sys
+
+
+
+if __name__ == "__main__":
+	args = sys.argv
+
+	if len(args)!=8 :
+		print "Usage: ",  args[0] , " infile outfile red green blue T lambda"
+		sys.exit(0)
+
+	img = vigra.readImage(args[1])
+
+	if img.shape[2]!=3:
+		print "Image must be RGB"
+		sys.exit(0)
+
+	T 	 = float(args[6])
+	beta = float(args[7])
+
+	
+
+	imgFlat = img.reshape([-1,3]).view(numpy.ndarray)
+	numVar  = imgFlat.shape[0]
+
+
+	gm = opengm.gm(numpy.ones(numVar,dtype=opengm.label_type)*2)
+
+	protoColor = numpy.array([args[3],args[4],args[5]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor = numpy.repeat(protoColor,numVar,axis=1).swapaxes(0,1)
+	diffArray  = numpy.sum(numpy.abs(imgFlat - protoColor),axis=1)
+	unaries    = numpy.ones([numVar,2],dtype=opengm.value_type)
+	unaries[:,0]=T
+	unaries[:,1]=diffArray
+
+	print diffArray
+
+	gm.addFactors(gm.addFunctions(unaries),numpy.arange(numVar))
+
+
+	regularizer=opengm.pottsFunction([2,2],0.0,beta)
+	gridVariableIndices=opengm.secondOrderGridVis(img.shape[0],img.shape[1])
+
+	fid=gm.addFunction(regularizer)
+	gm.addFactors(fid,gridVariableIndices)
+
+	print gm
+
+	inf=opengm.inference.GraphCut(gm)
+	inf.infer()
+	arg=inf.arg().reshape(img.shape[0:2])
+
+	vigra.impex.writeImage(args[2])
\ No newline at end of file
diff --git a/src/tutorials/python/demo/demo2.py b/src/tutorials/python/demo/demo2.py
new file mode 100644
index 0000000..2b2ac9c
--- /dev/null
+++ b/src/tutorials/python/demo/demo2.py
@@ -0,0 +1,56 @@
+import opengm
+import vigra
+import numpy
+import sys
+
+
+
+if __name__ == "__main__":
+	args = sys.argv
+
+	if len(args)!=8 :
+		print "Usage: ",  args[0] , " infile outfile red green blue T lambda"
+		sys.exit(0)
+
+	img = vigra.readImage(args[1])
+
+	if img.shape[2]!=3:
+		print "Image must be RGB"
+		sys.exit(0)
+
+	T 	 = float(args[6])
+	beta = float(args[7])
+
+	
+
+	imgFlat = img.reshape([-1,3]).view(numpy.ndarray)
+	numVar  = imgFlat.shape[0]
+
+
+	gm = opengm.gm(numpy.ones(numVar,dtype=opengm.label_type)*2)
+
+	protoColor = numpy.array([args[3],args[4],args[5]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor = numpy.repeat(protoColor,numVar,axis=1).swapaxes(0,1)
+	diffArray  = numpy.sum(numpy.abs(imgFlat - protoColor),axis=1)
+	unaries    = numpy.ones([numVar,2],dtype=opengm.value_type)
+	unaries[:,0]=T
+	unaries[:,1]=diffArray
+
+	print diffArray
+
+	gm.addFactors(gm.addFunctions(unaries),numpy.arange(numVar))
+
+
+	regularizer=opengm.pottsFunction([2,2],0.0,beta)
+	gridVariableIndices=opengm.secondOrderGridVis(img.shape[0],img.shape[1])
+
+	fid=gm.addFunction(regularizer)
+	gm.addFactors(fid,gridVariableIndices)
+
+	print gm
+
+	inf=opengm.inference.GraphCut(gm)
+	inf.infer()
+	arg=inf.arg().reshape(img.shape[0:2])
+
+	vigra.impex.writeImage(arg,args[2])
\ No newline at end of file
diff --git a/src/tutorials/python/demo/demo3.py b/src/tutorials/python/demo/demo3.py
new file mode 100644
index 0000000..944cd90
--- /dev/null
+++ b/src/tutorials/python/demo/demo3.py
@@ -0,0 +1,50 @@
+import opengm
+import vigra
+import numpy
+import sys
+
+
+
+if __name__ == "__main__":
+	args = sys.argv
+
+	if len(args)!=8 :
+		print "Usage: ",  args[0] , " infile hdf5-outfile red green blue T lambda"
+		sys.exit(0)
+
+	img = vigra.readImage(args[1])
+
+	if img.shape[2]!=3:
+		print "Image must be RGB"
+		sys.exit(0)
+
+	T 	 = float(args[6])
+	beta = float(args[7])
+
+	
+
+	imgFlat = img.reshape([-1,3]).view(numpy.ndarray)
+	numVar  = imgFlat.shape[0]
+
+
+	gm = opengm.gm(numpy.ones(numVar,dtype=opengm.label_type)*2)
+
+	protoColor = numpy.array([args[3],args[4],args[5]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor = numpy.repeat(protoColor,numVar,axis=1).swapaxes(0,1)
+	diffArray  = numpy.sum(numpy.abs(imgFlat - protoColor),axis=1)
+	unaries    = numpy.ones([numVar,2],dtype=opengm.value_type)
+	unaries[:,0]=T
+	unaries[:,1]=diffArray
+
+	print diffArray
+
+	gm.addFactors(gm.addFunctions(unaries),numpy.arange(numVar))
+
+
+	regularizer=opengm.pottsFunction([2,2],0.0,beta)
+	gridVariableIndices=opengm.secondOrderGridVis(img.shape[0],img.shape[1])
+
+	fid=gm.addFunction(regularizer)
+	gm.addFactors(fid,gridVariableIndices)
+
+	opengm.hdf5.saveGraphicalModel(gm,args[2],"gm")
\ No newline at end of file
diff --git a/src/tutorials/python/demo/demo4.py b/src/tutorials/python/demo/demo4.py
new file mode 100644
index 0000000..f0a3405
--- /dev/null
+++ b/src/tutorials/python/demo/demo4.py
@@ -0,0 +1,62 @@
+import opengm
+import vigra
+import numpy
+import sys
+
+
+
+if __name__ == "__main__":
+	args = sys.argv
+
+	if len(args)!=11 :
+		print "Usage: ",  args[0] , " infile outfile red1 green1 blue1 red2 green2 blue2 T lambda"
+		sys.exit(0)
+
+	img = vigra.readImage(args[1])
+
+	if img.shape[2]!=3:
+		print "Image must be RGB"
+		sys.exit(0)
+
+	T 	 = float(args[9])
+	beta = float(args[10])
+
+	
+
+	imgFlat = img.reshape([-1,3]).view(numpy.ndarray)
+	numVar  = imgFlat.shape[0]
+
+
+	gm = opengm.gm(numpy.ones(numVar,dtype=opengm.label_type)*3)
+
+	protoColor1 = numpy.array([args[3],args[4],args[5]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor1 = numpy.repeat(protoColor1,numVar,axis=1).swapaxes(0,1)
+
+	protoColor2 = numpy.array([args[6],args[7],args[8]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor2 = numpy.repeat(protoColor2,numVar,axis=1).swapaxes(0,1)
+
+	diffArray1 = numpy.sum(numpy.abs(imgFlat - protoColor1),axis=1)
+	diffArray2 = numpy.sum(numpy.abs(imgFlat - protoColor2),axis=1)
+	unaries    = numpy.ones([numVar,3],dtype=opengm.value_type)
+	unaries[:,0]=T
+	unaries[:,1]=diffArray1
+	unaries[:,2]=diffArray2
+
+
+	gm.addFactors(gm.addFunctions(unaries),numpy.arange(numVar))
+
+
+	regularizer=opengm.pottsFunction([3,3],0.0,beta)
+	gridVariableIndices=opengm.secondOrderGridVis(img.shape[0],img.shape[1])
+
+	fid=gm.addFunction(regularizer)
+	gm.addFactors(fid,gridVariableIndices)
+
+	if True :
+		inf=opengm.inference.TrwsExternal(gm)
+	else:
+		inf=opengm.inference.BeliefPropagation(gm)
+
+	inf.infer()
+	arg=inf.arg().reshape(img.shape[0:2])
+	vigra.impex.writeImage(arg,args[2])
\ No newline at end of file
diff --git a/src/tutorials/python/demo/demo5.py b/src/tutorials/python/demo/demo5.py
new file mode 100644
index 0000000..b05b70b
--- /dev/null
+++ b/src/tutorials/python/demo/demo5.py
@@ -0,0 +1,92 @@
+import opengm
+import vigra
+import numpy
+import sys
+
+import matplotlib
+from matplotlib import pyplot as plt
+
+if __name__ == "__main__":
+	args = sys.argv
+
+	if len(args)!=11 :
+		print "Usage: ",  args[0] , " infile outfile red1 green1 blue1 red2 green2 blue2 T lambda"
+		sys.exit(0)
+
+	img = vigra.readImage(args[1])
+
+	if img.shape[2]!=3:
+		print "Image must be RGB"
+		sys.exit(0)
+
+	T 	 = float(args[9])
+	beta = float(args[10])
+
+	
+
+	imgFlat = img.reshape([-1,3]).view(numpy.ndarray)
+	numVar  = imgFlat.shape[0]
+
+
+	gm = opengm.gm(numpy.ones(numVar,dtype=opengm.label_type)*3)
+
+	protoColor1 = numpy.array([args[3],args[4],args[5]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor1 = numpy.repeat(protoColor1,numVar,axis=1).swapaxes(0,1)
+
+	protoColor2 = numpy.array([args[6],args[7],args[8]],dtype=opengm.value_type).reshape([3,-1])
+	protoColor2 = numpy.repeat(protoColor2,numVar,axis=1).swapaxes(0,1)
+
+	diffArray1 = numpy.sum(numpy.abs(imgFlat - protoColor1),axis=1)
+	diffArray2 = numpy.sum(numpy.abs(imgFlat - protoColor2),axis=1)
+	unaries    = numpy.ones([numVar,3],dtype=opengm.value_type)
+	unaries[:,0]=T
+	unaries[:,1]=diffArray1
+	unaries[:,2]=diffArray2
+
+
+	gm.addFactors(gm.addFunctions(unaries),numpy.arange(numVar))
+
+
+	regularizer=opengm.pottsFunction([3,3],0.0,beta)
+	gridVariableIndices=opengm.secondOrderGridVis(img.shape[0],img.shape[1])
+
+	fid=gm.addFunction(regularizer)
+	gm.addFactors(fid,gridVariableIndices)
+
+	visitNth=1
+	if False :
+		visitNth=1000
+		inf=opengm.inference.Icm(gm)
+	elif False:
+		inf=opengm.inference.TrwsExternal(gm)
+	else:
+		inf=opengm.inference.BeliefPropagation(gm,parameter=opengm.InfParam(damping=0.9))
+
+	class PyCallback(object):
+	    def __init__(self,shape,numLabels):
+	        self.shape=shape
+	        self.numLabels=numLabels
+	        self.cmap = matplotlib.colors.ListedColormap ( numpy.random.rand ( self.numLabels,3))
+	        matplotlib.interactive(True)
+	    def begin(self,inference):
+	        print "begin of inference"
+	    def end(self,inference):
+	        print "end of inference"
+	    def visit(self,inference):
+	    	print "visit"
+	        gm=inference.gm()
+	        labelVector=inference.arg()
+	        print labelVector.shape
+	        print self.shape
+	        labelVector=labelVector.reshape(self.shape)
+	        plt.imshow(labelVector*255.0, cmap=self.cmap,interpolation="nearest") 
+	       	plt.draw()
+
+
+	callback=PyCallback(img.shape[0:2],3)
+	visitor=inf.pythonVisitor(callback,visitNth=visitNth)
+
+	#inf.infer(inf.verboseVisitor())
+	inf.infer(visitor)
+	arg=inf.arg().reshape(img.shape[0:2])
+	vigra.impex.writeImage(arg,args[2])
\ No newline at end of file

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/opengm.git