def __init__(self,s,r) :
self.s = s
self.r = r
self.f = mlab.figure()
self.e3d = []
rect_disparity, mrf_params, dummy = disparity_map(rect_from, rect_to, disprange)
if rectified:
disparity = rect_disparity
else:
disparity = unrectify_image(rect_disparity, H, epipole, w, h, rect_table, emptyflag = rect_disparity.min(), samplingfunc = nearest_color)
# Model building
f = K[0, 0]
b = alg.norm(P1[:, 3] - P2[:, 3])
Z = f * b / (disparity + b * 0.5)
Y, X = mgrid[0:h, 0:w]
X = X * Z / f
Y = Y * Z / f
from enthought.tvtk.tools import mlab
mlab.figure(browser = False).add(mlab.Surf(X, Y, Z, image_from))
#mlab.figure(browser = False).add(mlab.ImShow(disparity, show_scalar_bar = True))
#from scipy.io.matlab import mio
#mio.savemat('disparity.mat', {'disparity': disparity, 'F_ft': F_ft, 'rect_from': rect_from, 'rect_to': rect_to, 'Z': Z, 'X': X, 'Y': Y, 'image': image_from})
#MATLAB: warp(X, Y, -Z, image)
#draw_image_features(image_from, xy1, xy_from)
#draw_image_features(image_to, xy2, xy_to)
#draw_image_epipolar(image_from, xy_to, F_ft, H, (minx, miny))
#draw_image_epipolar(image_to, xy_from, F_ft.T)
#draw_image(rect_from)
#draw_image(rect_to)
#draw_image(disparity)
#
# <codecell>
import scipy
# prepare some interesting function:
def f(x, y):
return 3.0*scipy.sin(x*y+1e-4)/(x*y+1e-4)
x = scipy.arange(-7., 7.05, 0.1)
y = scipy.arange(-5., 5.05, 0.1)
# 3D visualization of f:
from enthought.tvtk.tools import mlab
fig = mlab.figure()
s = mlab.SurfRegular(x, y, f)
fig.add(s)
# <markdowncell>
# mlab_attachments/simple_example.png
#
# Changing axis and colour
# ========================
#
# <codecell>
from scipy import *
import scipy
# prepare some interesting function:
def f(x, y):
return 3.0*scipy.sin(x*y+1e-4)/(x*y+1e-4)
x = scipy.arange(-7., 7.05, 0.1)
y = scipy.arange(-5., 5.05, 0.1)
# 3D visualization of f:
from enthought.tvtk.tools import mlab
fig = mlab.figure()
s = mlab.SurfRegular(x, y, f)
fig.add(s)
w,
h,
rect_table,
emptyflag=rect_disparity.min(),
samplingfunc=nearest_color)
# Model building
f = K[0, 0]
b = alg.norm(P1[:, 3] - P2[:, 3])
Z = f * b / (disparity + b * 0.5)
Y, X = mgrid[0:h, 0:w]
X = X * Z / f
Y = Y * Z / f
from enthought.tvtk.tools import mlab
mlab.figure(browser=False).add(mlab.Surf(X, Y, Z, image_from))
#mlab.figure(browser = False).add(mlab.ImShow(disparity, show_scalar_bar = True))
#from scipy.io.matlab import mio
#mio.savemat('disparity.mat', {'disparity': disparity, 'F_ft': F_ft, 'rect_from': rect_from, 'rect_to': rect_to, 'Z': Z, 'X': X, 'Y': Y, 'image': image_from})
#MATLAB: warp(X, Y, -Z, image)
#draw_image_features(image_from, xy1, xy_from)
#draw_image_features(image_to, xy2, xy_to)
#draw_image_epipolar(image_from, xy_to, F_ft, H, (minx, miny))
#draw_image_epipolar(image_to, xy_from, F_ft.T)
#draw_image(rect_from)
#draw_image(rect_to)
#draw_image(disparity)
