Selaa lähdekoodia

Fix for flake8

Kentaro Wada 7 vuotta sitten
vanhempi
commit
6ed9ee7ee7
3 muutettua tiedostoa jossa 7 lisäystä ja 6 poistoa
  1. 2 1
      labelme/cli/draw_json.py
  2. 1 1
      labelme/cli/json_to_dataset.py
  3. 4 4
      labelme/lib.py

+ 2 - 1
labelme/cli/draw_json.py

@@ -1,10 +1,11 @@
 #!/usr/bin/env python
 
 import argparse
+import base64
 import json
 import os
 import sys
-import base64
+
 import matplotlib.pyplot as plt
 
 from labelme import utils

+ 1 - 1
labelme/cli/json_to_dataset.py

@@ -1,8 +1,8 @@
 import argparse
+import base64
 import json
 import os
 import os.path as osp
-import base64
 import warnings
 
 import PIL.Image

+ 4 - 4
labelme/lib.py

@@ -1,6 +1,6 @@
 from math import sqrt
 import os.path as osp
-from numpy.linalg import norm
+
 import numpy as np
 
 from qtpy import QtCore
@@ -76,10 +76,10 @@ def distancetoline(point, line):
     p2 = np.array([p2.x(), p2.y()])
     p3 = np.array([point.x(), point.y()])
     if np.dot((p3 - p1), (p2 - p1)) < 0:
-        return norm(p3 - p1)
+        return np.linalg.norm(p3 - p1)
     if np.dot((p3 - p2), (p1 - p2)) < 0:
-        return norm(p3 - p2)
-    return norm(np.cross(p2 - p1, p1 - p3)) / norm(p2 - p1)
+        return np.linalg.norm(p3 - p2)
+    return np.linalg.norm(np.cross(p2 - p1, p1 - p3)) / np.linalg.norm(p2 - p1)
 
 
 def fmtShortcut(text):