Exemplo n.º 1
0
    def __init__(self, res=(640,480), Nframes=10):
        x=np.linspace(-10,10, res[0])
        y=np.linspace(-10,10, res[1])
        Y,X=np.meshgrid(y,x)
        Ypos=np.linspace(-3,3,Nframes/2)
        Ypos=np.hstack([Ypos, Ypos[::-1]])
        #sclX=ones(Nframes)
        #gauss2d=100*gauss(X*sclX[:,np.newaxis, np.newaxis], [1, 0,0.2])*gauss(Y, [1, 0,0.2])
        self.dat=np.empty((Nframes, res[0], res[1]))
        for k in range(Nframes):
            if k<4:
                g2d=0
            else:
                g2d=200*MT.gauss2d(X,Y, 4, 4+0*Ypos[k],0.5,0.5,0)
            self.dat[k]=1*abs(np.random.normal(size=(res[0], res[1]) ) ) + g2d        #self.dat=self.dat.astype('u4')

        self.nextInd=0
Exemplo n.º 2
0
    y, x = np.mgrid[:frame.shape[0], :frame.shape[1]]
    m00 = frame.sum() #or 1.0
    m10 = (frame * x).sum() / m00
    m01 = (frame * y).sum() / m00
    dx, dy = x - m10, y - m01
    m20 = (frame * dx ** 2).sum() / m00
    m02 = (frame * dy ** 2).sum() / m00
    m11 = (frame * dx * dy).sum() / m00


    q = N.sqrt((m20 - m02) ** 2 + 4 * m11 ** 2)
    minor_axis = 2 ** 1.5 * N.sqrt(m20 + m02 + q)
    major_axis = 2 ** 1.5 * N.sqrt(m20 + m02 - q)
    angle = 0.5 * N.arctan2(2 * m11, m20 - m02)
    ellipticity = minor_axis / major_axis

    centroid = (m01, m10)

    return 8*m00/major_axis/minor_axis/N.pi, centroid, major_axis/4, minor_axis/4, angle

if __name__=="__main__":
    from numpy import *
    from numpy import random
    import MT
    #x=arange(640)
    #y=arange(480)
    X,Y=mgrid[:640, :480]
    frame=7*MT.gauss2d(X,Y, 320, 240, 50,70, 0)+0.0*random.normal(size=(640,480))
    gfit=GaussFitter()
    #print(gfit.fitgaussian(frame))