Function coco_pipe

dali/test/python/reader/test_coco.py:257–264

(coco_op, file_root, annotations_file, polygon_masks, pixelwise_masks)

@pipeline_def(batch_size=batch_size_alias_test, device_id=0, num_threads=4)
def coco_pipe(coco_op, file_root, annotations_file, polygon_masks, pixelwise_masks):
    inputs, boxes, labels, *_ = coco_op(
        file_root=file_root,
        annotations_file=annotations_file,
        polygon_masks=polygon_masks,
        pixelwise_masks=pixelwise_masks,
    )
    return inputs, boxes, labels


def test_coco_reader_alias():