Export YOLO with filters in new sdk

[pirnerjonas]

Hi all,

I found https://github.com/HumanSignal/label-studio-sdk/blob/master/examples/export_with_filters.py which shows how you export annotations with filters but it uses the old version of the sdk from label_studio_sdk import Client.

The export example for the new sdk in the readme only shows this:

annotations = [
    task.annotations
    for task in ls.tasks.list(project=project.id, fields='all')
    if task.annotations
]

How is it possible to filter tasks and then export the tasks annotations to YOLO with the new sdk?

[heidi-humansignal]

Hello,

Do you mind trying this:

from label_studio_sdk.client import LabelStudio from label_studio_sdk.data_manager import Filters, Column, Operator, Type from label_studio_sdk.converter import Converter import json # 1. Connect ls = LabelStudio(base_url="https://YOUR_LABEL_STUDIO_URL", api_key="YOUR_API_KEY") project_id = YOUR_PROJECT_ID # 2. Build your filter (example: file name contains "cat") filters = Filters.create( Filters.AND, [Filters.item( Column.data("filename"), Operator.CONTAINS, Type.String, Filters.value("cat"), )] ) # 3. List filtered tasks query = json.dumps({"filters": filters}) tasks = ls.tasks.list(project=project_id, fields="all", query=query) tasks_list = list(tasks) print(f"Found {len(tasks_list)} tasks") # 4. Convert to YOLO project_info = ls.projects.retrieve(project_id) config_xml = project_info.label_config converter = Converter( config=config_xml, project_dir=None, upload_dir=None, download_resources=False, ) # it requires a check, maybe this function is not implemented converter.convert(tasks_list, to_format="YOLO", output_dir="./yolo_export")

Thank you,
Abu

Comment by Abubakar Saad
Workflow Run

[pirnerjonas]

I had to change .retrieve() to .get() and converter.convert() has args input_data, output_data and format.

https://github.com/HumanSignal/label-studio-sdk/blob/master/src/label_studio_sdk/converter/converter.py#L234-L319

and it looks to me like input_data has to be either a dir or a json.

[pirnerjonas]

I also found https://github.com/HumanSignal/label-studio-sdk/blob/master/examples/export_with_filters.py

but to be clear I don't want to be able to only filter based on tasks basis but also be able to filter certain annotations of a task based on a condition. E.g. get me all polygon label annotations where a choice label is yes.

And after that filter I want to be able to flexible export to e.g. YOLO or COCO.