Skip to content

Commit 82466e6

Browse files
vpandiarajan20danielbotros
authored andcommitted
[DATA-4604]Classifications not showing up in dataset.jsonl (viamrobotics#5254)
1 parent e321acd commit 82466e6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

cli/dataset.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -317,6 +317,10 @@ func binaryDataToJSONLines(ctx context.Context, client datapb.DataServiceClient,
317317
for _, tag := range datum.GetMetadata().GetCaptureMetadata().GetTags() {
318318
annotations = append(annotations, Annotation{AnnotationLabel: tag})
319319
}
320+
classificationsAnnotations := datum.GetMetadata().GetAnnotations().GetClassifications()
321+
for _, classification := range classificationsAnnotations {
322+
annotations = append(annotations, Annotation{AnnotationLabel: classification.GetLabel()})
323+
}
320324
bboxAnnotations := convertBoundingBoxes(datum.GetMetadata().GetAnnotations().GetBboxes())
321325

322326
fileName := filepath.Join(dst, dataDir, filenameForDownload(datum.GetMetadata()))

0 commit comments

Comments
 (0)