FEAT: Adding vision functions

Author: pavanky

arrayfire/__init__.py

@@ -19,3 +19,4 @@
 from .signal     import *
 from .image      import *
 from .features   import *
+from .vision     import *

arrayfire/vision.py

@@ -0,0 +1,39 @@
+#######################################################
+# Copyright (c) 2015, ArrayFire
+# All rights reserved.
+#
+# This file is distributed under 3-clause BSD license.
+# The complete license agreement can be obtained at:
+# http://arrayfire.com/licenses/BSD-3-Clause
+########################################################
+from .library import *
+from .array import *
+from .features import *
+
+def fast(image, threshold=20.0, arc_length=9, non_max=True, feature_ratio=0.05, edge=3):
+    out = features()
+    safe_call(clib.af_fast(pointer(out.feat),\
+                           image.arr, c_float(threshold), c_uint(arc_length), non_max, \
+                           c_float(feature_ratio), c_uint(edge)))
+    return out
+
+def orb(image, threshold=20.0, max_features=400, scale = 1.5, num_levels = 4, blur_image = False):
+    feat = features()
+    desc = array()
+    safe_call(clib.af_orb(pointer(feat.feat), pointer(desc.arr),\
+                          c_float(threshold), c_uint(max_features),\
+                          c_float(scale), c_uint(num_levels), blur_image))
+    return feat, desc
+
+def hamming_matcher(query, database, dim = 0, num_nearest = 1):
+    index = array()
+    dist = array()
+    safe_call(clib.af_hamming_matcher(pointer(idx.arr), pointer(dist.arr),\
+                                      query.arr, database.arr, \
+                                      c_longlong(dim), c_longlong(num_nearest)))
+    return index, dist
+
+def match_template(image, template, match_type = AF_SAD):
+    out = array()
+    safe_call(clib.af_match_template(pointer(out.arr), image.arr, template.arr, match_type))
+    return out