correct code tomake pylsd working

primetang · Jun 29, 2016 · b796b0f · b796b0f · melissa-bei · Sep 1, 2020
1 parent 9a89672
commit b796b0f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/example/example_cv2.py b/example/example_cv2.py
@@ -14,7 +14,7 @@
 src = cv2.imread(fullName, cv2.IMREAD_COLOR)
 gray = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
 lines = lsd(gray)
-for i in xrange(lines.shape[0]):
+for i in range(lines.shape[0]):
     pt1 = (int(lines[i, 0]), int(lines[i, 1]))
     pt2 = (int(lines[i, 2]), int(lines[i, 3]))
     width = lines[i, 4]

diff --git a/pylsd/bindings/lsd_ctypes.py b/pylsd/bindings/lsd_ctypes.py
@@ -31,7 +31,7 @@ def load_lsd_library():
             try:
                 lsdlib = ctypes.cdll[os.path.join(root_dir, libdir, libname)]
                 return lsdlib
-            except Exception, e:
+            except Exception as e:
                 pass
         tmp = os.path.dirname(root_dir)
         if tmp == root_dir:

diff --git a/pylsd/lsd.py b/pylsd/lsd.py
@@ -19,10 +19,10 @@ def lsd(src):
     src = (ctypes.c_double * lens)(*src)
     lsdlib.lsdGet(src, ctypes.c_int(rows), ctypes.c_int(cols), temp)
 
-    fp = open(temp, 'r')
+    fp = open('D', 'r')
     cnt = fp.read().strip().split(' ')
     fp.close()
-    os.remove(temp)
+    os.remove('D')
 
     count = int(cnt[0])
     dim = int(cnt[1])