diff --git a/next_cvat/app/download.py b/next_cvat/app/download.py index bc7ed6c..8dad705 100644 --- a/next_cvat/app/download.py +++ b/next_cvat/app/download.py @@ -25,6 +25,11 @@ def download( "-f", help="Load credentials from a specific .env file", ), + include_images: bool = typer.Option( + True, + "--include-images", + help="Include images in the dataset", + ), ): """ Download annotations and images from a CVAT project. @@ -35,4 +40,6 @@ def download( username=settings_.username, password=settings_.password, token=settings_.token, - ).download_(project_id=project_id, dataset_path=dataset_path) + ).download_( + project_id=project_id, dataset_path=dataset_path, include_images=include_images + ) diff --git a/next_cvat/client/client.py b/next_cvat/client/client.py index 885e730..2600515 100644 --- a/next_cvat/client/client.py +++ b/next_cvat/client/client.py @@ -85,8 +85,10 @@ def list_projects(self): def project(self, project_id: int) -> Project: return Project(client=self, id=project_id) - def download_(self, project_id, dataset_path): - return self.project(project_id=project_id).download_(dataset_path) + def download_(self, project_id, dataset_path, include_images=True): + return self.project(project_id=project_id).download_( + dataset_path=dataset_path, include_images=include_images + ) Project.model_rebuild() diff --git a/next_cvat/client/project.py b/next_cvat/client/project.py index 3def388..be56b1d 100644 --- a/next_cvat/client/project.py +++ b/next_cvat/client/project.py @@ -29,7 +29,7 @@ def cvat(self) -> Generator[CVATProject, None, None]: with self.client.cvat_client() as client: yield client.projects.retrieve(self.id) - def download_(self, dataset_path: Union[str, Path]) -> Project: + def download_(self, dataset_path: Union[str, Path], include_images=True) -> Project: """Download project data to the specified path. Args: @@ -49,7 +49,7 @@ def download_(self, dataset_path: Union[str, Path]) -> Project: cvat_project.export_dataset( format_name="CVAT for images 1.1", filename=temp_file_path, - include_images=True, + include_images=include_images, ) # Extract contents