Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import cv2
+import json
 
 from pyAAMED import pyAAMED
 
@@ -12,13 +13,37 @@ def detect_ellipses(img_path):
     imgC = cv2.imread(img_path)
     imgG = cv2.cvtColor(imgC, cv2.COLOR_BGR2GRAY)
 
-    aamed = pyAAMED(600, 600)
-
+    ammed_size = 1200
+
+    iheight, iwidth, _ = imgG.shape
+    imax = max(iheight, iwidth)
+    iscale = ammed_size / imax
+    is_landscape = iwidth >= iheight
+    if is_landscape:
+        iw = imax * iscale
+        ih = iheight * iscale
+    else:
+        iw = iwidth * iscale
+        ih = imax * iscale
+
+    imgG = imgG.resize(iw, ih)
+
+    if is_landscape:
+        ipad = (imax - iheight) / 2 * iscale
+        imgG = cv2.copyMakeBorder(imgG, ipad, ipad, 0, 0, cv2.BORDER_REPLICATE)
+    else:
+        ipad = (imax - iwidth) / 2 * iscale
+        imgG = cv2.copyMakeBorder(imgG, 0, 0, ipad, ipad, cv2.BORDER_REPLICATE)
+
+    aamed = pyAAMED(ammed_size, ammed_size)
     aamed.setParameters(3.1415926/3, 3.4, 0.77)
+    
     result = aamed.run_AAMED(imgG)
+    result = result.replace(' ', ', ')
+    result = json.loads(result)
     print(result)
 
-    return [img_path, result]
+    return [img_path, json.dumps(result)]
 
 examples = [
     ["./AAMED/python/002_0038.jpg"]