Get the model settings based on the provided parameters or default values. Args: use_doc_orientation_classify (Union[bool, None]): Enables document orientation classification if True. Defaults to system setting if None. use_doc_unwarping (Union[bool, None]):
(
self,
use_doc_orientation_classify: Union[bool, None],
use_doc_unwarping: Union[bool, None],
use_layout_detection: Union[bool, None],
use_chart_recognition: Union[bool, None],
format_block_content: Union[bool, None],
merge_layout_blocks: Union[bool, None],
markdown_ignore_labels: Optional[list[str]] = None,
)
| 206 | ) |
| 207 | |
| 208 | def get_model_settings( |
| 209 | self, |
| 210 | use_doc_orientation_classify: Union[bool, None], |
| 211 | use_doc_unwarping: Union[bool, None], |
| 212 | use_layout_detection: Union[bool, None], |
| 213 | use_chart_recognition: Union[bool, None], |
| 214 | format_block_content: Union[bool, None], |
| 215 | merge_layout_blocks: Union[bool, None], |
| 216 | markdown_ignore_labels: Optional[list[str]] = None, |
| 217 | ) -> dict: |
| 218 | """ |
| 219 | Get the model settings based on the provided parameters or default values. |
| 220 | |
| 221 | Args: |
| 222 | use_doc_orientation_classify (Union[bool, None]): Enables document orientation classification if True. Defaults to system setting if None. |
| 223 | use_doc_unwarping (Union[bool, None]): Enables document unwarping if True. Defaults to system setting if None. |
| 224 | |
| 225 | Returns: |
| 226 | dict: A dictionary containing the model settings. |
| 227 | |
| 228 | """ |
| 229 | if use_doc_orientation_classify is None and use_doc_unwarping is None: |
| 230 | use_doc_preprocessor = self.use_doc_preprocessor |
| 231 | else: |
| 232 | if use_doc_orientation_classify is True or use_doc_unwarping is True: |
| 233 | use_doc_preprocessor = True |
| 234 | else: |
| 235 | use_doc_preprocessor = False |
| 236 | |
| 237 | if use_layout_detection is None: |
| 238 | use_layout_detection = self.use_layout_detection |
| 239 | |
| 240 | if use_chart_recognition is None: |
| 241 | use_chart_recognition = self.use_chart_recognition |
| 242 | |
| 243 | if format_block_content is None: |
| 244 | format_block_content = self.format_block_content |
| 245 | |
| 246 | if merge_layout_blocks is None: |
| 247 | merge_layout_blocks = self.merge_layout_blocks |
| 248 | |
| 249 | if markdown_ignore_labels is None: |
| 250 | markdown_ignore_labels = self.markdown_ignore_labels |
| 251 | |
| 252 | return dict( |
| 253 | use_doc_preprocessor=use_doc_preprocessor, |
| 254 | use_layout_detection=use_layout_detection, |
| 255 | use_chart_recognition=use_chart_recognition, |
| 256 | format_block_content=format_block_content, |
| 257 | merge_layout_blocks=merge_layout_blocks, |
| 258 | markdown_ignore_labels=markdown_ignore_labels, |
| 259 | ) |
| 260 | |
| 261 | def check_model_settings_valid(self, input_params: dict) -> bool: |
| 262 | """ |