fjenett commited on
Commit
90c0555
·
1 Parent(s): e49c811

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,7 +16,7 @@ def detect_ellipses(img_path):
16
 
17
  ammed_size = 1200
18
 
19
- iheight, iwidth = np.array(imgG.shape).astype('int')
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)