Fix bug: The truth value of an array with more than one element is am…

…biguous. (yolov8-seg)
egbertYeah · Mar 11, 2024 · 1074458 · 1074458
1 parent e1151b7
commit 1074458
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/anylabeling/services/auto_labeling/__base__/yolo.py b/anylabeling/services/auto_labeling/__base__/yolo.py
@@ -314,7 +314,7 @@ def predict_shapes(self, image, image_path=None):
                 shape.label = self.classes[int(class_id)]
                 shape.selected = False
                 shapes.append(shape)
-            if self.task == "seg" and point:
+            if self.task == "seg":
                 shape = Shape(flags={})
                 for p in point:
                     shape.add_point(QtCore.QPointF(int(p[0]), int(p[1])))