def setup(self, count=1, fn=None):
self._count = count
self._heights = []
self.blob_size = 100
if fn is None:
self.process_fn = lambda r, g, b: (g - r).threshold(30).erode()
else:
self.process_fn = fn
self._get_centroids = _get_centroids(count)
# setup variables
for i in range(count):
stream = Stream(title="Height %s" % (i + 1), type=int)
setattr(self, "height%s" % (i + 1), stream)
self._heights.append(stream)
self.image = Image(title="Tracked", fn=self._get_image)
self.visualisation = DerivedImage(title="Visualisation")
self.status = Property(title="Status", type=str)
self.ui = ui(properties=[self.status, self.image, self.visualisation] +
self._heights)
def setup (self, count = 1, fn = None): self._count = count self._heights = [] self.blob_size = 100 if fn is None: self.process_fn = lambda r, g, b: (g - r).threshold(30).erode() else: self.process_fn = fn self._get_centroids = _get_centroids(count) # setup variables for i in range(count): stream = Stream(title = "Height %s" % (i + 1), type = int) setattr(self, "height%s" % (i + 1), stream) self._heights.append(stream) self.image = Image(title = "Tracked", fn = self._get_image) self.visualisation = DerivedImage(title = "Visualisation") self.status = Property(title = "Status", type = str) self.ui = ui( properties = [self.status, self.image, self.visualisation] + self._heights )
def setup(self, fn=None):
# setup variables
self.height = Stream(title="Height", type=int)
self.status = Property(title="Status", type=str)
self.image = Image(title="Tracked", fn=self._get_image)
self.visualisation = DerivedImage(title="Visualisation")
if fn is None:
self.process_fn = lambda r, g, b: (g - r).threshold(30).erode()
else:
self.process_fn = fn
self.blob_size = 100
self._get_centroids = _get_centroids(1)
self.ui = ui(properties=[
self.status, self.height, self.image, self.visualisation
])
def setup (self, fn = None): # setup variables self.height = Stream(title = "Height", type = int) self.status = Property(title = "Status", type = str) self.image = Image(title = "Tracked", fn = self._get_image) self.visualisation = DerivedImage(title = "Visualisation") if fn is None: self.process_fn = lambda r, g, b: (g - r).threshold(30).erode() else: self.process_fn = fn self.blob_size = 100 self._get_centroids = _get_centroids(1) self.ui = ui( properties = [self.status, self.height, self.image, self.visualisation] )
class MultiBlobTracker (Machine):
protocolFactory = None
name = "Follow Blobs on a webcam"
x_tolerance = 30
update_frequency = 1
def setup (self, count = 1, fn = None):
self._count = count
self._heights = []
self.blob_size = 100
if fn is None:
self.process_fn = lambda r, g, b: (g - r).threshold(30).erode()
else:
self.process_fn = fn
self._get_centroids = _get_centroids(count)
# setup variables
for i in range(count):
stream = Stream(title = "Height %s" % (i + 1), type = int)
setattr(self, "height%s" % (i + 1), stream)
self._heights.append(stream)
self.image = Image(title = "Tracked", fn = self._get_image)
self.visualisation = DerivedImage(title = "Visualisation")
self.status = Property(title = "Status", type = str)
self.ui = ui(
properties = [self.status, self.image, self.visualisation] + self._heights
)
def start (self):
img = self.protocol.image()
img = self.process_fn(*img.splitChannels())
pos = self._get_centroids(img, self.blob_size)
self._x = []
if len(pos) < self._count:
raise Exception("Could not find %s blobs" % self._count)
for p in pos:
self._x.append(p[0])
self._tick(self.image.refresh, self.update_frequency)
def show (self):
self.image.value.show()
def _get_image (self):
img = self.protocol.image()
if img is None:
return None
processed_img = self.process_fn(*img.splitChannels())
self.visualisation._push(processed_img)
pos = self._get_centroids(processed_img, self.blob_size)
if pos is not None:
for x, y in pos:
img.drawRectangle(x - 10, y - 10, 20, 20, (255,) * 3, width = 6)
found = 0
for i in range(self._count):
_x = self._x[i]
for x, y in pos:
if abs(x - _x) < self.x_tolerance:
self._heights[i]._push(img.height - y)
found += 1
if found < self._count:
self.status._push("blobs-missing")
else:
self.status._push("ok")
else:
self.status._push("error")
return img
def stop (self):
self._stopTicks()
def disconnect (self):
self.stop()
try:
self.protocol.disconnect()
except AttributeError:
pass
class SingleBlobTracker (Machine):
protocolFactory = None
name = "Follow Green Blob on a webcam"
update_frequency = 1
def setup (self, fn = None):
# setup variables
self.height = Stream(title = "Height", type = int)
self.status = Property(title = "Status", type = str)
self.image = Image(title = "Tracked", fn = self._get_image)
self.visualisation = DerivedImage(title = "Visualisation")
if fn is None:
self.process_fn = lambda r, g, b: (g - r).threshold(30).erode()
else:
self.process_fn = fn
self.blob_size = 100
self._get_centroids = _get_centroids(1)
self.ui = ui(
properties = [self.status, self.height, self.image, self.visualisation]
)
def start (self):
self._tick(self.image.refresh, self.update_frequency)
def show (self):
self.image.value.show()
def _get_image (self):
img = self.protocol.image()
if img is None:
return
processed_img = self.process_fn(*img.splitChannels())
self.visualisation._push(processed_img)
pos = self._get_centroids(processed_img, self.blob_size)
if pos is not None:
x, y = pos[0]
img.drawRectangle(x - 10, y - 10, 20, 20, (255,) * 3, width = 6)
self.height._push(img.height - pos[0][1])
self.status._push("ok")
else:
self.status._push("error")
return img
def stop (self):
self._stopTicks()
def disconnect (self):
self.stop()
try:
self.protocol.disconnect()
except AttributeError:
pass
class MultiBlobTracker(Machine):
protocolFactory = None
name = "Follow Blobs on a webcam"
x_tolerance = 30
update_frequency = 1
def setup(self, count=1, fn=None):
self._count = count
self._heights = []
self.blob_size = 100
if fn is None:
self.process_fn = lambda r, g, b: (g - r).threshold(30).erode()
else:
self.process_fn = fn
self._get_centroids = _get_centroids(count)
# setup variables
for i in range(count):
stream = Stream(title="Height %s" % (i + 1), type=int)
setattr(self, "height%s" % (i + 1), stream)
self._heights.append(stream)
self.image = Image(title="Tracked", fn=self._get_image)
self.visualisation = DerivedImage(title="Visualisation")
self.status = Property(title="Status", type=str)
self.ui = ui(properties=[self.status, self.image, self.visualisation] +
self._heights)
def start(self):
img = self.protocol.image()
img = self.process_fn(*img.splitChannels())
pos = self._get_centroids(img, self.blob_size)
self._x = []
if len(pos) < self._count:
raise Exception("Could not find %s blobs" % self._count)
for p in pos:
self._x.append(p[0])
self._tick(self.image.refresh, self.update_frequency)
def show(self):
self.image.value.show()
def _get_image(self):
img = self.protocol.image()
if img is None:
return None
processed_img = self.process_fn(*img.splitChannels())
self.visualisation._push(processed_img)
pos = self._get_centroids(processed_img, self.blob_size)
if pos is not None:
for x, y in pos:
img.drawRectangle(x - 10, y - 10, 20, 20, (255, ) * 3, width=6)
found = 0
for i in range(self._count):
_x = self._x[i]
for x, y in pos:
if abs(x - _x) < self.x_tolerance:
self._heights[i]._push(img.height - y)
found += 1
if found < self._count:
self.status._push("blobs-missing")
else:
self.status._push("ok")
else:
self.status._push("error")
return img
def stop(self):
self._stopTicks()
def disconnect(self):
self.stop()
try:
self.protocol.disconnect()
except AttributeError:
pass
class SingleBlobTracker(Machine):
protocolFactory = None
name = "Follow Green Blob on a webcam"
update_frequency = 1
def setup(self, fn=None):
# setup variables
self.height = Stream(title="Height", type=int)
self.status = Property(title="Status", type=str)
self.image = Image(title="Tracked", fn=self._get_image)
self.visualisation = DerivedImage(title="Visualisation")
if fn is None:
self.process_fn = lambda r, g, b: (g - r).threshold(30).erode()
else:
self.process_fn = fn
self.blob_size = 100
self._get_centroids = _get_centroids(1)
self.ui = ui(properties=[
self.status, self.height, self.image, self.visualisation
])
def start(self):
self._tick(self.image.refresh, self.update_frequency)
def show(self):
self.image.value.show()
def _get_image(self):
img = self.protocol.image()
if img is None:
return
processed_img = self.process_fn(*img.splitChannels())
self.visualisation._push(processed_img)
pos = self._get_centroids(processed_img, self.blob_size)
if pos is not None:
x, y = pos[0]
img.drawRectangle(x - 10, y - 10, 20, 20, (255, ) * 3, width=6)
self.height._push(img.height - pos[0][1])
self.status._push("ok")
else:
self.status._push("error")
return img
def stop(self):
self._stopTicks()
def disconnect(self):
self.stop()
try:
self.protocol.disconnect()
except AttributeError:
pass