From 2bcd1e1dfdab345f3cb095a872894ed93a7641ab Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Tue, 18 Apr 2023 20:36:52 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=9B=A0=EF=B8=8Fsmall=20fixes?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/__init__.py | 1 + supervision/dataset/core.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) 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: