added support for yolo instance-segmentation polygon format

mfl28 · Mar 4, 2024 · dd87cb8 · dd87cb8
1 parent 35f37a7
commit dd87cb8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/github/mfl28/boundingboxeditor/model/io/YOLOLoadStrategy.java b/src/main/java/com/github/mfl28/boundingboxeditor/model/io/YOLOLoadStrategy.java
@@ -163,9 +163,9 @@ private ImageAnnotation loadAnnotationFromFile(File file) throws IOException {
                     int lineCoordinatesCount = Arrays.asList(StringUtils.split(line)).size() - 1;
 
                     try {
-                        if(lineCoordinatesCount == 4){
+                        if(lineCoordinatesCount <= 4){
                             boundingShapeDataList.add(parseBoundingBoxData(line, counter));
-                        } else if(lineCoordinatesCount > 4){
+                        } else {
                             boundingShapeDataList.add(parseBoundingPolygonData(line, counter));
                         }
                     } catch(InvalidAnnotationFormatException e) {