Das deutsche Python-Forum

Code: Alles auswählen

    def houghtf_lin(self, img, angles=None):
        """Perform the straight line Hough transform.
    
        Input:
          img - a boolean array
          angles - in degrees
    
        Output:
          H - the Hough transform coefficients
          distances
          angles
        
        """
        
        if img.ndim != 2:
            raise ValueError("Input must be a two-dimensional array")
    
        img = img.astype(bool)
        
        if not angles:
            angles = np.linspace(-90,90,180)
    
        theta = angles / 180. * np.pi
        d = np.ceil(np.hypot(*img.shape))
        nr_bins = 2*d - 1
        bins = np.linspace(-d,d,nr_bins)
        out = np.zeros((nr_bins,len(theta)),dtype=itype)
    
        rows,cols = img.shape
        x,y = np.mgrid[:rows,:cols]
    
        for i,(cT,sT) in enumerate(zip(np.cos(theta),np.sin(theta))):
            rho = np.round_(cT*x[img] + sT*y[img]) - bins[0] + 1
            rho = rho.astype(itype)
            rho[(rho < 0) | (rho > nr_bins)] = 0
            bc = np.bincount(rho.flat)[1:]
            out[:len(bc),i] = bc
    
        return out,angles,bins

Code: Alles auswählen

        
        im = Image.fromarray(img)
        cv_im = cv.CreateImageHeader(im.size, cv.IPL_DEPTH_8U, 3)
        cv.SetData(cv_im, im.tostring())
        
        gray = cv.CreateImage(cv.GetSize(cv_im), 8, 1)
        storage = cv.CreateMat(cv_im.width, 1, cv.CV_32FC3)

        circles = cv.HoughCircles(gray, storage, cv.CV_HOUGH_GRADIENT,1,1, 200, 100) 

        print np.asarray(storage)

Code: Alles auswählen

import cv, numpy

mat = cv.CreateMat(3, 5, cv.CV_32FC3)
print type(mat)

cv.Set(mat, 7)
a = numpy.asarray(mat)
print a