Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions luxonis_ml/data/parsers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
from .segmentation_mask_directory_parser import SegmentationMaskDirectoryParser
from .solo_parser import SOLOParser
from .tensorflow_csv_parser import TensorflowCSVParser
from .ultralytics_parser import UltralyticsParser
from .voc_parser import VOCParser
from .yolov4_parser import YoloV4Parser
from .yolov6_parser import YoloV6Parser
Expand All @@ -21,6 +22,7 @@
"SOLOParser",
"SegmentationMaskDirectoryParser",
"TensorflowCSVParser",
"UltralyticsParser",
"VOCParser",
"YoloV4Parser",
"YoloV6Parser",
Expand Down
2 changes: 2 additions & 0 deletions luxonis_ml/data/parsers/luxonis_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
from .segmentation_mask_directory_parser import SegmentationMaskDirectoryParser
from .solo_parser import SOLOParser
from .tensorflow_csv_parser import TensorflowCSVParser
from .ultralytics_parser import UltralyticsParser
from .voc_parser import VOCParser
from .yolov4_parser import YoloV4Parser
from .yolov6_parser import YoloV6Parser
Expand All @@ -48,6 +49,7 @@ class LuxonisParser(Generic[T]):
DatasetType.SEGMASK: SegmentationMaskDirectoryParser,
DatasetType.SOLO: SOLOParser,
DatasetType.NATIVE: NativeParser,
DatasetType.ULTRALYTICS: UltralyticsParser,
}

def __init__(
Expand Down
Loading
Loading