~bertrand-nouvel/pycvf-keypoints/trunk

# -*- coding: utf-8 -*-

from pycvf.core import genericnode

from pycvfext.image.datatypes import image

from pycvf.datatypes import list as ldt

 

import numpy

from pycvfext.keypoints.lib.gb.compute_channels_oe_nms import compute_channels_oe_nms

from pycvfext.keypoints.lib.gb.get_descriptors import get_descriptors_at

 

class GeometricBlurDescriptors:

  def __init__(self,

                      blurRate=0.5,

                      blurBase=1,

                      sampleRadii=[0, 4, 8, 16, 32, 50],

                      numSamplePerRadius= [1, 8, 8, 10, 12, 12],

                      usePBEdges=False):

        self.usePBEdges=usePBEdges

        self.blurRate=blurRate

        self.blurBase=blurBase

        self.sampleRadii=sampleRadii

        self.numSamplePerRadius=numSamplePerRadius

  def process(self,args):

     I,keypoints=args

     if I.ndim==3:

       if (I.shape[2]>1):

         I = I.mean(axis=2)

       else:

         I=I.reshape(I.shape[:2])

    

     if not self.usePBEdges:

       fbr = compute_channels_oe_nms(I)

     else:

       assert False,"Not yet implemented"

     descrs, frames=get_descriptors_at(numpy.dstack(fbr), keypoints, self.blurRate, self.blurBase, numpy.array(self.sampleRadii), numpy.array(self.numSamplePerRadius))

     

 

Node=genericnode.pycvf_node_class(image.Datatype(),None)(GeometricBlurDescriptors)

__call__=Node