Author: Hakan Ardo <[email protected]>
Branch: extradoc
Changeset: r4809:cbf97e9be316
Date: 2012-09-13 17:23 +0200
http://bitbucket.org/pypy/extradoc/changeset/cbf97e9be316/
Log: renumber labels for nicer intermediate video
diff --git a/talk/dls2012/demo/detect.py b/talk/dls2012/demo/detect.py
--- a/talk/dls2012/demo/detect.py
+++ b/talk/dls2012/demo/detect.py
@@ -54,6 +54,15 @@
def __getitem__(self, (x, y)):
return self.labels[x, y]
+ def renumber(self):
+ ll = list(set(self.labels))
+ ll.sort()
+ if ll[0] != 0:
+ ll.insert(0, 0)
+ for x, y in self.labels.indexes():
+ self.labels[x, y] = ll.index(self.labels[x, y])
+ self.last_label = len(ll) - 1
+
def bwlabel(seg):
labels = Labler(seg)
@@ -68,7 +77,7 @@
if seg[x, y]:
ll = [labels[x, y], labels[x+1, y], labels[x-1, y+1],
labels[x, y+1], labels[x+1, y+1]]
labels.update(x, y, ll)
-
+ labels.renumber()
return labels.labels
class BoundingBox(object):
@@ -109,5 +118,5 @@
labels = bwlabel(seg)
boxes = extract_boxes(labels)
boxes = [b for b in boxes if b.area() >= minarea]
- view(64*labels, 'segments')
+ viewsc(labels, 'segments')
return boxes
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
