class IntensityTracker(IntensityTrackedBodypart):
def __init__(self,
name=None,
params={},
color='white',
bEqualizeHist=False):
IntensityTrackedBodypart.__init__(self, name, params, color,
bEqualizeHist)
self.state = MsgState()
self.state.intensity = 0.0
self.state.freq = 0.0
self.wingbeat = WingbeatDetector(0, 1000)
self.set_params(params)
# set_params()
# Set the given params dict into this object.
#
def set_params(self, params):
IntensityTrackedBodypart.set_params(self, params)
self.imgRoiBackground = None
self.iCount = 0
self.state.intensity = 0.0
self.wingbeat.set(self.params['wingbeat_min'],
self.params['wingbeat_max'])
# update_state()
#
def update_state(self):
#f = 175 # Simulate this freq.
#t = gImageTime
#self.state.intensity = np.cos(2*np.pi*f*t)
self.state.intensity = np.sum(self.imgRoiFgMasked).astype(
np.float32) / self.mask.sum
self.state.freq = self.wingbeat.freq_from_intensity(
self.state.intensity, 1.0 / self.dt)
# update()
# Update all the internals given a foreground camera image.
#
def update(self, dt, image, bInvertColor):
IntensityTrackedBodypart.update(self, dt, image, bInvertColor)
if (self.params['gui'][self.name]['track']):
self.update_state()
# draw()
# Draw the outline.
#
def draw(self, image):
IntensityTrackedBodypart.draw(self, image)
class IntensityTracker(IntensityTrackedBodypart):
def __init__(self, name=None, params={}, color='white', bEqualizeHist=False):
IntensityTrackedBodypart.__init__(self, name, params, color, bEqualizeHist)
self.state = MsgState()
self.state.intensity = 0.0
self.state.freq = 0.0
self.wingbeat = WingbeatDetector(0, 1000)
self.set_params(params)
# set_params()
# Set the given params dict into this object.
#
def set_params(self, params):
IntensityTrackedBodypart.set_params(self, params)
self.imgRoiBackground = None
self.iCount = 0
self.state.intensity = 0.0
self.wingbeat.set(self.params['wingbeat_min'],
self.params['wingbeat_max'])
# update_state()
#
def update_state(self):
#f = 175 # Simulate this freq.
#t = gImageTime
#self.state.intensity = np.cos(2*np.pi*f*t)
self.state.intensity = np.sum(self.imgRoiFgMasked).astype(np.float32) / self.mask.sum
self.state.freq = self.wingbeat.freq_from_intensity(self.state.intensity, 1.0/self.dt)
# update()
# Update all the internals given a foreground camera image.
#
def update(self, dt, image, bInvertColor):
IntensityTrackedBodypart.update(self, dt, image, bInvertColor)
if (self.params['gui'][self.name]['track']):
self.update_state()
# draw()
# Draw the outline.
#
def draw(self, image):
IntensityTrackedBodypart.draw(self, image)
def __init__(self,
name=None,
params={},
color='white',
bEqualizeHist=False):
IntensityTrackedBodypart.__init__(self, name, params, color,
bEqualizeHist)
self.state = MsgState()
self.state.intensity = 0.0
self.state.freq = 0.0
self.wingbeat = WingbeatDetector(0, 1000)
self.set_params(params)
def __init__(self, name=None, params={}, color='white', bEqualizeHist=False):
IntensityTrackedBodypart.__init__(self, name, params, color, bEqualizeHist)
self.state = MsgState()
self.state.intensity = 0.0
self.state.freq = 0.0
self.wingbeat = WingbeatDetector(0, 1000)
self.set_params(params)
