diff --git a/supervision/__init__.py b/supervision/__init__.py index 7adbd41d..49348033 100644 --- a/supervision/__init__.py +++ b/supervision/__init__.py @@ -8,6 +8,7 @@ from supervision.detection.tools.polygon_zone import PolygonZone, PolygonZoneAnn from supervision.detection.utils import generate_2d_mask, mask_to_xyxy from supervision.draw.color import Color, ColorPalette from supervision.draw.utils import draw_filled_rectangle, draw_polygon, draw_text +from supervision.file import list_files_with_extensions from supervision.geometry.core import Point, Position, Rect from supervision.geometry.utils import get_polygon_center from supervision.notebook.utils import plot_image, plot_images_grid diff --git a/supervision/dataset/core.py b/supervision/dataset/core.py index b6b4f187..ff33c6a3 100644 --- a/supervision/dataset/core.py +++ b/supervision/dataset/core.py @@ -26,8 +26,8 @@ class Dataset: def as_pascal_voc( self, - images_directory_path: Optional[str], - annotations_directory_path: Optional[str], + images_directory_path: Optional[str] = None, + annotations_directory_path: Optional[str] = None, minimum_detection_area_percentage: float = 0.0, maximum_detection_area_percentage: float = 1.0, ) -> None: