diff --git a/exp/hough.py b/exp/hough.py --- a/exp/hough.py +++ b/exp/hough.py @@ -59,11 +59,13 @@ class HoughTransform: log.debug("alpha, beta: %s, %s",alpha,beta) self._drawLine(img,alpha,beta,peaks,i) + res.append([]) keys=self._readLineKeys(alpha,beta) for k in peaks: (alphaDeg,d)=keys[k] line=Line(alphaDeg*math.pi/180,d-self._diagLen//2) - res.append(self._transformOutput(line)) + res[-1].append(self._transformOutput(line)) + res[-1].sort(key=lambda line: line.d) i+=1 self.show(img)