Fix bug: list index out of range for exporting dota annotations

CVHub520 · Apr 22, 2024 · 755c578 · 755c578
1 parent cd422f5
commit 755c578
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/anylabeling/views/labeling/label_converter.py b/anylabeling/views/labeling/label_converter.py
@@ -733,8 +733,11 @@ def custom_to_dota(self, input_file, output_file):
             data = json.load(f)
         with open(output_file, "w", encoding="utf-8") as f:
             for shape in data["shapes"]:
-                label = shape["label"]
                 points = shape["points"]
+                shape_type = shape["shape_type"]
+                if shape_type != "rotation" or len(points) != 4:
+                    continue
+                label = shape["label"]
                 difficult = shape.get("difficult", False)
                 x0 = points[0][0]
                 y0 = points[0][1]