데이터셋에 포함된 내용 이해하기

1 분

이 페이지는 데이터셋에 포함된 내용을 이해하는 데 도움을 줍니다.

데이터셋에 어떤 씬에 대한 어노테이션이 있는지 확인하기

어떤 씬에 대해 예측을 제공할 수 있는지 알기 위해서는, 데이터셋에 어떤 씬에 대한 어노테이션이 있는지 먼저 확인해야 합니다. 이는 데이터셋 내 어노테이션 목록을 조회하는 엔드포인트를 사용하여 확인할 수 있습니다. 현재는 어노테이션을 데이터셋에 추가함으로써만 간접적으로 씬을 데이터셋에 추가할 수 있습니다. 이 작업에 도움이 필요하거나 씬을 데이터셋에 연결하고 싶다면 문의해 주십시오. 아래 코드 스니펫은 이 엔드포인트를 활용하여 데이터를 CSV 파일로 저장합니다.

from kognic.auth.requests.auth_session import RequestsAuthSession
import csv

base_url = "https://dataset.app.kognic.com"
client = RequestsAuthSession()
dataset_uuid = YOUR_DATASET_UUID
done = False
offset = 0
annotations = []
while not done:
    print(f"Fetched {offset} annotations")
    res = client.session.get(
        f"{base_url}/v1/datasets/{dataset_uuid}/annotations?offset={offset}"
    )
    res_annotations = res.json()['data']
    annotations += res_annotations
    offset += len(res_annotations)
    if len(res_annotations) == 0:
        done = True
print(f"Fetched {len(annotations)} annotations")

# save to csv called dataset_{dataset_uuid}_annotations.csv
with open(f'dataset_{dataset_uuid}_annotations.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(annotations[0].keys())
    for annotation in annotations:
        writer.writerow(annotation.values())