Small changes to Python samples

    Update find_obj.py:
    Added additional error checking for the loaded files and modified the detector check to be more pythonic (is versus ==).

    Update common.py:
    Fixed a typo (contais -> contains)

diff --git a/samples/python2/common.py b/samples/python2/common.py
index 4c6f800c44be376eb8c69a5264295f009f5f27af..12cef7c2c75deeb018ef340aaf21381781bcd8eb 100755 (executable)
--- a/samples/python2/common.py
+++ b/samples/python2/common.py
@@ -1,7 +1,7 @@
 #/usr/bin/env python
 
 '''
-This module contais some common routines used by other samples.
+This module contains some common routines used by other samples.
 '''
 
 import numpy as np

diff --git a/samples/python2/find_obj.py b/samples/python2/find_obj.py
index 7a875ca4f77cbc2cd1fafc73dbea0ee3ede5e849..1b28a7aae42f81805687c3b3a4339bb9edc97e81 100755 (executable)
--- a/samples/python2/find_obj.py
+++ b/samples/python2/find_obj.py
@@ -9,7 +9,7 @@ USAGE
   --feature  - Feature to use. Can be sift, surf of orb. Append '-flann' to feature name
                 to use Flann-based matcher instead bruteforce.
 
-  Press left mouse button on a feature point to see its mathcing point.
+  Press left mouse button on a feature point to see its matching point.
 '''
 
 import numpy as np
@@ -130,7 +130,8 @@ if __name__ == '__main__':
     opts, args = getopt.getopt(sys.argv[1:], '', ['feature='])
     opts = dict(opts)
     feature_name = opts.get('--feature', 'sift')
-    try: fn1, fn2 = args
+    try:
+        fn1, fn2 = args
     except:
         fn1 = '../c/box.png'
         fn2 = '../c/box_in_scene.png'
@@ -138,12 +139,20 @@ if __name__ == '__main__':
     img1 = cv2.imread(fn1, 0)
     img2 = cv2.imread(fn2, 0)
     detector, matcher = init_feature(feature_name)
-    if detector != None:
-        print 'using', feature_name
-    else:
+
+    if img1 is None:
+        print 'Failed to load fn1:', fn1
+        sys.exit(1)
+        
+    if img2 is None:
+        print 'Failed to load fn2:', fn2
+        sys.exit(1)
+    
+    if detector is None:
         print 'unknown feature:', feature_name
         sys.exit(1)
-
+    
+    print 'using', feature_name
 
     kp1, desc1 = detector.detectAndCompute(img1, None)
     kp2, desc2 = detector.detectAndCompute(img2, None)