Function annotation_export_core

shared/export/export_generation.py:151–287

Generic method to export a file list

(
    session,
    project,
    export,
    file_list)

def annotation_export_core(
    session,
    project,
    export,
    file_list):
    """
    Generic method to export a file list
    """

    images_dir = f"{settings.PROJECT_IMAGES_BASE_DIR + str(project.id)}/"

    export.file_list_length = len(file_list)

    errors_result = check_for_errors(
        export = export,
        session = session)
    if errors_result is False:
        return False, None

    # If we build annotations directly then we could return them
    # If tf records than not
    # But some clean up stuff (ie marking complete) that's joint
    # Also not clear where we would be using a return dict of annotations here
    # Ohhh it returns annotations since we upload in YAML or JSON format for that
    # Maybe that should just be part of that process (instead of returning with a
    # seperate flag?)

    annotations = None

    # This is here as it's shared with with masks and
    # not masks, but needs to run before masks if masks

    # So we can have mask values increase in series
    # instead of using ids for example

    # Careful, want to use project default directory for labels for now

    label_file_list = WorkingDirFileLink.file_list(
        session = session,
        working_dir_id = export.project.directory_default_id,
        limit = None,
        type = "label"
    )

    if export.kind == "TF Records":
        label_dict = data_tools.label_dict_builder(
            file_list = label_file_list)

    export_label_map = {}
    for label_file in label_file_list:
        export_label_map[label_file.id] = label_file.label.name

    # TODO masks if not part of TF records is not really handled great right now
    # TODO pass export object to track it?

    """

        Would be good to allow masks for regular records / JSON