Detr Partitioner#

class sycamore.transforms.detr_partitioner.DeformableDetr(model_name_or_path, device=None)[source]#

Bases: SycamoreObjectDetection

infer(image: <module 'PIL.Image' from '/home/docs/checkouts/readthedocs.org/user_builds/sycamore/envs/stable/lib/python3.11/site-packages/PIL/Image.py'>, threshold: float) list[Element][source]#

Do inference using the wrapped model.

class sycamore.transforms.detr_partitioner.PDFMinerExtractor[source]#

Bases: object

class sycamore.transforms.detr_partitioner.SycamoreObjectDetection[source]#

Bases: ABC

Wrapper class for the various object detection models.

abstract infer(image: <module 'PIL.Image' from '/home/docs/checkouts/readthedocs.org/user_builds/sycamore/envs/stable/lib/python3.11/site-packages/PIL/Image.py'>, threshold: float) List[Element][source]#

Do inference using the wrapped model.

class sycamore.transforms.detr_partitioner.SycamorePDFPartitioner(model_name_or_path, device=None)[source]#

Bases: object