Skip to content

Commit 6ee08f3

Browse files
committed
Correct mapping of yolov8 subtypes to parsers
1 parent 234995a commit 6ee08f3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

luxonis_ml/data/parsers/luxonis_parser.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@ class LuxonisParser(Generic[T]):
5050
DatasetType.SOLO: SOLOParser,
5151
DatasetType.NATIVE: NativeParser,
5252
DatasetType.YOLOV8BOUNDINGBOX: YOLOv8Parser,
53+
DatasetType.YOLOV8INSTANCESEGMENTATION: YOLOv8Parser,
54+
DatasetType.YOLOV8KEYPOINTS: YOLOv8Parser,
5355
}
5456

5557
def __init__(

0 commit comments

Comments
 (0)