blenderproc.python.modules.writer.CocoAnnotationsWriter module¶

class blenderproc.python.modules.writer.CocoAnnotationsWriter.CocoAnnotationsWriter(config)[source]¶

Writes Coco Annotations in to a file.

Configuration:

Parameter	Description	Type
rgb_output_key	The output key with which the rgb images were registered. Should be the same as the output_key of the RgbRenderer module. Default: colors.	string
segmap_output_key	The output key with which the segmentation images were registered. Should be the same as the output_key of the SegMapRenderer module. Default: segmap.	string
segcolormap_output_key	The output key with which the csv file for object name/class correspondences was registered. Should be the same as the colormap_output_key of the SegMapRenderer module. Default: segcolormap.	string
supercategory	Name of the dataset/supercategory to filter for, e.g. a specific BOP dataset. Default: coco_annotations	string
append_to_existing_output	If true and if there is already a coco_annotations.json file in the output directory, the new coco annotations will be appended to the existing file. Also the rgb images will be named such that there are no collisions. Default: False.	bool
mask_encoding_format	Encoding format of the binary masks. Default: ‘rle’. Available: ‘rle’, ‘polygon’.	string

run()[source]¶: Writes coco annotations in the following steps: 1. Locat the seg images 2. Locat the rgb maps 3. Locat the seg maps 4. Read color mappings 5. For each frame write the coco annotation