Update postprocessing/dula/layout.py

postprocessing/dula/layout.py

@@ -44,9 +44,9 @@ def fit_layout(floor_xz, need_cube=False, show=False, block_eps=0.2):
     polys = cv2.findContours(floorplan_sub, 1, 2)
     if isinstance(polys, tuple):
         if len(polys) == 3:
-            polys = polys[1]
+            polys = list(polys[1])
         else:
-            polys = polys[0]
+            polys = list(polys[0])
     poly = polys[0]
     epsilon = 0.005 * cv2.arcLength(poly, True)
     poly = cv2.approxPolyDP(poly, epsilon, True)
@@ -150,9 +150,9 @@ def fit_layout(floor_xz, need_cube=False, show=False, block_eps=0.2):
     pred_polys = cv2.findContours(pred, 1, 3)
     if isinstance(pred_polys, tuple):
         if len(pred_polys) == 3:
-            pred_polys = pred_polys[1]
+            pred_polys = list(pred_polys[1])
         else:
-            pred_polys = pred_polys[0]
+            pred_polys = list(pred_polys[0])
 
     pred_polys.sort(key=lambda x: cv2.contourArea(x), reverse=True)
     pred_polys = pred_polys[0]