def __init__(self, **kwargs):
params = KcfParameters()
params.padding = kwargs.get('padding', 1.7)
params.lambdaValue = kwargs.get('lambdaValue', 0.0001)
params.outputSigmaFactor = kwargs.get('outputSigmaFactor', 0.05)
params.votScaleStep = kwargs.get('votScaleStep', 1.05)
params.votScaleWeight = kwargs.get('votScaleWeight', 0.95)
params.templateSize = kwargs.get('templateSize', 100)
params.interpFactor = kwargs.get('interpFactor', 0.012)
params.kernelSigma = kwargs.get('kernelSigma', 0.6)
params.cellSize = kwargs.get('cellSize', 4)
params.pixelPadding = kwargs.get('pixelPadding', 0)
params.enableTrackingLossDetection = kwargs.get('enableTrackingLossDetection', False)
params.psrThreshold = kwargs.get('psrThreshold', 13.5)
params.psrPeakDel = kwargs.get('psrPeakDel', 1)
params.useVotScaleEstimation = kwargs.get('useVotScaleEstimation', False)
params.useDsstScaleEstimation = kwargs.get('useDsstScaleEstimation', True)
params.scaleSigmaFactor = kwargs.get('scaleSigmaFactor', 0.25)
params.scaleEstimatorStep = kwargs.get('scaleEstimatorStep', 1.02)
params.scaleLambda = kwargs.get('scaleLambda', 0.01)
params.scaleCellSize = kwargs.get('caleCellSize', 4)
params.numberOfScales = kwargs.get('numberOfScales', 33)
params.resizeType = kwargs.get('resizeType', 1)
params.useFhogTranspose = kwargs.get('useFhogTranspose', False)
params.minArea = kwargs.get('minArea', 10)
params.maxAreaFactor = kwargs.get('maxAreaFactor', 0.8)
params.nScalesVot = kwargs.get('nScalesVot', 3)
params.votMinScaleFactor = kwargs.get('votMinScaleFactor', 0.01)
params.votMaxScaleFactor = kwargs.get('votMaxScaleFactor', 40)
params.useCcs = kwargs.get('useCcs', True)
self.params = params
self.tracker = KcfTracker(self.params)
self.initialized = False
class KCF:
def __init__(self, **kwargs):
params = KcfParameters()
params.padding = kwargs.get('padding', 1.7)
params.lambdaValue = kwargs.get('lambdaValue', 0.0001)
params.outputSigmaFactor = kwargs.get('outputSigmaFactor', 0.05)
params.votScaleStep = kwargs.get('votScaleStep', 1.05)
params.votScaleWeight = kwargs.get('votScaleWeight', 0.95)
params.templateSize = kwargs.get('templateSize', 100)
params.interpFactor = kwargs.get('interpFactor', 0.012)
params.kernelSigma = kwargs.get('kernelSigma', 0.6)
params.cellSize = kwargs.get('cellSize', 4)
params.pixelPadding = kwargs.get('pixelPadding', 0)
params.enableTrackingLossDetection = kwargs.get('enableTrackingLossDetection', False)
params.psrThreshold = kwargs.get('psrThreshold', 13.5)
params.psrPeakDel = kwargs.get('psrPeakDel', 1)
params.useVotScaleEstimation = kwargs.get('useVotScaleEstimation', False)
params.useDsstScaleEstimation = kwargs.get('useDsstScaleEstimation', True)
params.scaleSigmaFactor = kwargs.get('scaleSigmaFactor', 0.25)
params.scaleEstimatorStep = kwargs.get('scaleEstimatorStep', 1.02)
params.scaleLambda = kwargs.get('scaleLambda', 0.01)
params.scaleCellSize = kwargs.get('caleCellSize', 4)
params.numberOfScales = kwargs.get('numberOfScales', 33)
params.resizeType = kwargs.get('resizeType', 1)
params.useFhogTranspose = kwargs.get('useFhogTranspose', False)
params.minArea = kwargs.get('minArea', 10)
params.maxAreaFactor = kwargs.get('maxAreaFactor', 0.8)
params.nScalesVot = kwargs.get('nScalesVot', 3)
params.votMinScaleFactor = kwargs.get('votMinScaleFactor', 0.01)
params.votMaxScaleFactor = kwargs.get('votMaxScaleFactor', 40)
params.useCcs = kwargs.get('useCcs', True)
self.params = params
self.tracker = KcfTracker(self.params)
self.initialized = False
def init(self, image, boundingBox):
success = self.tracker.reinit(image, boundingBox)
if success:
self.initialized = True
else:
raise Exception('Unable to initialize KCF tracker with given frame.')
def reinit(self, image, boundingBox):
self.tracker.reinit(image, boundingBox)
def update(self, image):
return self.tracker.update(image)
def updateAt(self, image, boundingBox):
return self.tracker.updateAt(image, boundingBox)
def getBoundingBox(self):
return self.tracker.getBoundingBox()
def getCenter(self):
return self.tracker.getCenter()
