Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def detect_ellipses(img_path):
|
|
16 |
|
17 |
ammed_size = 1200
|
18 |
|
19 |
-
iheight, iwidth =
|
20 |
imax = max(iheight, iwidth)
|
21 |
iscale = ammed_size / imax
|
22 |
is_landscape = iwidth >= iheight
|
@@ -27,13 +27,13 @@ def detect_ellipses(img_path):
|
|
27 |
iw = iwidth * iscale
|
28 |
ih = imax * iscale
|
29 |
|
30 |
-
imgG = imgG.resize(iw, ih)
|
31 |
|
32 |
if is_landscape:
|
33 |
-
ipad = (imax - iheight) / 2 * iscale
|
34 |
imgG = cv2.copyMakeBorder(imgG, ipad, ipad, 0, 0, cv2.BORDER_REPLICATE)
|
35 |
else:
|
36 |
-
ipad = (imax - iwidth) / 2 * iscale
|
37 |
imgG = cv2.copyMakeBorder(imgG, 0, 0, ipad, ipad, cv2.BORDER_REPLICATE)
|
38 |
|
39 |
aamed = pyAAMED(ammed_size, ammed_size)
|
|
|
16 |
|
17 |
ammed_size = 1200
|
18 |
|
19 |
+
iheight, iwidth = imgG.shape
|
20 |
imax = max(iheight, iwidth)
|
21 |
iscale = ammed_size / imax
|
22 |
is_landscape = iwidth >= iheight
|
|
|
27 |
iw = iwidth * iscale
|
28 |
ih = imax * iscale
|
29 |
|
30 |
+
imgG = imgG.resize(int(iw), int(ih))
|
31 |
|
32 |
if is_landscape:
|
33 |
+
ipad = int((imax - iheight) / 2 * iscale)
|
34 |
imgG = cv2.copyMakeBorder(imgG, ipad, ipad, 0, 0, cv2.BORDER_REPLICATE)
|
35 |
else:
|
36 |
+
ipad = int((imax - iwidth) / 2 * iscale)
|
37 |
imgG = cv2.copyMakeBorder(imgG, 0, 0, ipad, ipad, cv2.BORDER_REPLICATE)
|
38 |
|
39 |
aamed = pyAAMED(ammed_size, ammed_size)
|