def __init__(self, org_id: str, data_set_name: str) -> None:
"""Construct Loader."""
self.org_id = org_id
self.data_set_name = data_set_name
self.api_client = RedBrickApi(cache=False)
print_info("Retrieving dataset ...")
# Dataset info
try:
dataset = self.api_client.get_datapointset(self.org_id, self.data_set_name)[
"dataPointSet"
]
except Exception as err:
print_error(err)
return
print_info("Dataset successfully retrieved!")
self.org_id = dataset["orgId"]
self.data_set_name = dataset["name"]
self.data_type = dataset["dataType"]
self.datapoint_count = dataset["datapointCount"]
self.desc = dataset["desc"]
self.createdAt = dataset["createdAt"]
self.createdBy = dataset["createdBy"]
self.status = dataset["status"]
def __init__(self, org_id: str, label_set_name: str) -> None:
"""Construct Loader."""
self.org_id = org_id
self.label_set_name = label_set_name
self.api_client = RedBrickApi(cache=False)
print_info("Counting available datapoints...")
# All datapoints in labelset
try:
self.dp_ids, custom_group = self.api_client.get_datapoint_ids(
self.org_id, self.label_set_name)
except Exception as err:
print_error(err)
return
self.task_type = custom_group.task_type
self.data_type = custom_group.data_type
self.taxonomy: Dict[str, int] = custom_group.taxonomy
print_info("Number of Datapoints = %s" % len(self.dp_ids))
# Update taxonomy mapper if segmentation
if self.task_type == "SEGMENTATION":
self.taxonomy_update_segmentation()
# Get all users
try:
self.users = self.api_client.get_members(self.org_id)
except Exception as err:
print_error(err)
return
def __init__(
self,
org_id: str,
data_set_name: str,
storage_id: str,
label_set_name: str = None,
):
"""Construct creator"""
super().__init__()
self.org_id = org_id
self.data_set_name = data_set_name
self.storage_id = storage_id
self.label_set_name = label_set_name
self.api_client = RedBrickApi(cache=False)
def __init__(self, org_id: str, label_set_name: str) -> None:
self.api = RedBrickApi()
self.cursor = None
self.datapointsBatch = None
self.datapointsBatchIndex = None
self.org_id = org_id
self.label_set_name = label_set_name
self.valid_task_types: List = ["SEGMENTATION", "POLYGON", "MULTI"]
self.customGroup = self._get_custom_group()
self.taxonomy = self.customGroup["taxonomy"]
self.datapointCount = self.customGroup["datapointCount"]
self.taxonomy_segm = self._get_taxonomy_segmentation()
def __init__(self, org_id: str, project_id: str, stage_name: str) -> None:
"""Construct RemoteLabel instance."""
print(colored("[INFO]:", "blue"),
"Initializing remote-labeling module...")
self.org_id = org_id
self.project_id = project_id
self.stage_name = stage_name
self.api_client = RedBrickApi(cache=False)
# Gather stage information and store
stage_info = self.api_client.get_stage(org_id=org_id,
project_id=project_id,
stage_name=stage_name)
taxonomy = self.api_client.get_taxonomy(
orgId=org_id,
name=stage_info["outputTaxonomyName"],
version=stage_info["outputTaxonomyVersion"],
)
self.taxonomy: Taxonomy2 = taxonomy
self.task_type = stage_info["outputType"]