class EdgeSnapWindow(Window):
def __init__(self):
super(EdgeSnapWindow, self).__init__('Color Segmentation Example')
self.image = Image("shapes.png", sample=True).edges()
self.points = []
self.show(self.image)
def on_mouse(self, event, x, y, mouse_key, data=None):
""" Callback for mouse events
event - int - see cv2.EVENT_* constants
x, y - int, int - position of the cursor
mouse_key - int - mouse key
"""
if event == cv2.EVENT_LBUTTONDOWN:
self.points.append((x, y))
self.image.clear_layers()
for p in self.points:
self.image.dl().circle(p, 5, color=Color.BLUE)
features = self.image.edge_snap(self.points)
for f in features:
f.draw(color=Color.RED, width=2)
self.show(self.image.apply_layers())
def on_key(self, key):
if key == 32: # Space bar to clear points
self.points = []
self.image.clear_layers()
self.show(self.image)
print "Points cleared"
class ColorSegmentationWindow(Window):
def __init__(self):
super(ColorSegmentationWindow, self).__init__('Color Segmentation Example')
self.img = Image('simplecv')
self.segmentation = ColorSegmentation()
self.normal = True # mode toggle for segment detection
self.point1 = (0, 0)
self.point2 = (0, 0)
self.mosue_down = False
def on_mouse(self, event, x, y, mouse_key, data=None):
""" Callback for mouse events
event - int - see cv2.EVENT_* constants
x, y - int, int - position of the cursor
mouse_key - int - mouse key
"""
if event == cv2.EVENT_LBUTTONDOWN:
self.point1 = (x, y)
self.point2 = (x, y)
self.mosue_down = True
elif event == cv2.EVENT_MOUSEMOVE:
if self.mosue_down == True:
self.point2 = (x, y)
elif event == cv2.EVENT_LBUTTONUP:
self.mosue_down = False
self.point2 = (x, y)
x1, y1, w, h = points_to_roi(self.point1, self.point2)
if w > 0 and h > 0:
crop = self.img.crop(x1, y1, w, h)
self.segmentation = ColorSegmentation()
self.segmentation.add_to_model(crop)
def on_key(self, key):
if key == 32: # Space bar to switch between modes
self.normal = not self.normal
print "Display Mode:", "Normal" if self.normal else "Segmented"
def on_update(self):
""" Callback for periodic update.
"""
if self.normal:
self.img.clear_layers()
self.img.dl().rectangle_to_pts(self.point1, self.point2, color=Color.RED)
self.show(self.img.apply_layers())
else:
self.segmentation.add_image(self.img)
if self.segmentation.is_ready():
img = self.segmentation.get_segmented_image()
img = img.erode(iterations = 2).dilate().invert()
self.show(img)
class EdgeSnapWindow(Window):
def __init__(self):
super(EdgeSnapWindow, self).__init__('Color Segmentation Example')
self.image = Image("shapes.png", sample=True).edges()
self.points = []
self.show(self.image)
def on_mouse(self, event, x, y, mouse_key, data=None):
""" Callback for mouse events
event - int - see cv2.EVENT_* constants
x, y - int, int - position of the cursor
mouse_key - int - mouse key
"""
if event == cv2.EVENT_LBUTTONDOWN:
self.points.append((x, y))
self.image.clear_layers()
for p in self.points:
self.image.dl().circle(p, 5, color=Color.BLUE)
features = self.image.edge_snap(self.points)
for f in features:
f.draw(color=Color.RED, width=2)
self.show(self.image.apply_layers())
def on_key(self, key):
if key == 32: # Space bar to clear points
self.points = []
self.image.clear_layers()
self.show(self.image)
print "Points cleared"
img = Image("color.jpg", sample=True)
line_l = DrawingLayer()
a = (20, 20)
b = (20, 100)
c = (100, 100)
d = (100, 20)
line_l.line(a, b, alpha=128, width=5)
line_l.line(b, c, alpha=128)
line_l.line(c, d, antialias=True)
line_l.line(d, a, color=Color.PUCE)
line_l.line(b, d, width=5)
img.add_drawing_layer(line_l)
temp = img.apply_layers()
print "line: %s" % temp.save(temp=True)
img.clear_layers()
lines_l = DrawingLayer()
a = (20, 20)
b = (20, 100)
c = (100, 100)
d = (100, 20)
pts = (a, b, c, d, a)
lines_l.lines(pts, alpha=128)
# translate over and down 10
pts = map(lambda x: ((x[0] + 10), (x[1] + 10)), pts)
lines_l.lines(pts, color=Color.BEIGE, width=10)
#translate over and down 10
pts = map(lambda x: ((x[0] + 10), (x[1] + 10)), pts)
lines_l.lines(pts, antialias=True)
img.add_drawing_layer(lines_l)