Method __call__

openocr.py:193–243

Execute the task with appropriate parameters. For 'det' task: Args: image_path: Path to image or directory return_mask: Whether to return detection mask For 'rec' task: Args: image_path: Path to image

(self, *args, **kwargs)

        self.model = OpenDocONNX(**kwargs)

    def __call__(self, *args, **kwargs):
        """
        Execute the task with appropriate parameters.

        For 'det' task:
            Args:
                image_path: Path to image or directory
                return_mask: Whether to return detection mask

        For 'rec' task:
            Args:
                image_path: Path to image or directory
                batch_num: Batch size for recognition

        For 'ocr' task:
            Args:
                image_path: Path to image or directory
                is_visualize: Whether to visualize results
                rec_batch_num: Batch size for recognition
                crop_infer: Whether to use crop inference
                return_mask: Whether to return detection mask

        For 'unirec' task:
            Args:
                image_path: Path to image or PDF file
                max_length: Maximum generation length

        For 'doc' task:
            Args:
                image_path: Path to image or PDF file
                layout_threshold: Layout detection threshold
                max_length: Maximum generation length
                merge_layout_blocks: Whether to merge layout blocks

        Returns:
            Task-specific results
        """
        if self.model is None:
            raise RuntimeError('Model not initialized')

        # Dispatch to appropriate task
        if self.task == 'det':
            return self._call_det(*args, **kwargs)
        elif self.task == 'rec':
            return self._call_rec(*args, **kwargs)
        elif self.task == 'ocr':
            return self._call_ocr(*args, **kwargs)
        elif self.task == 'unirec':
            return self._call_unirec(*args, **kwargs)
        elif self.task == 'doc':
            return self._call_doc(*args, **kwargs)

    def _call_det(self, image_path=None, **kwargs):
        """Call detection task"""