Преглед изворни кода

Support createMode="ai_polygon" in self.double_click

Kentaro Wada пре 2 година
родитељ
комит
8613cfcf4c
1 измењених фајлова са 3 додато и 1 уклоњено
  1. 3 1
      labelme/widgets/canvas.py

+ 3 - 1
labelme/widgets/canvas.py

@@ -547,7 +547,9 @@ class Canvas(QtWidgets.QWidget):
         if self.double_click != "close":
         if self.double_click != "close":
             return
             return
 
 
-        if self.createMode == "polygon" and self.canCloseShape():
+        if (
+            self.createMode == "polygon" and self.canCloseShape()
+        ) or self.createMode == "ai_polygon":
             self.finalise()
             self.finalise()
 
 
     def selectShapes(self, shapes):
     def selectShapes(self, shapes):