fiftyone.utils.yolo.YOLOAnnotationWriter

class documentation

class YOLOAnnotationWriter(object): (source)

Class for writing annotations in YOLO-style TXT format.

Method write Writes the labels to disk.

def write(self, label, txt_path, labels_map_rev, dynamic_classes=False, include_confidence=False): (source) ¶

Writes the labels to disk.

Parameters
label	a `fiftyone.core.labels.Detections` or `fiftyone.core.labels.Polylines`
txt_path	the path to write the annotation TXT file
labels_map_rev	a dictionary mapping class label strings to target integers
dynamic_classes:`False`	whether to dynamically add new labels to `labels_map_rev`
include_confidence:`False`	whether to include confidences in the export, if they exist