UniMERNet
Unimernet Math Extractor¶
In [ ]:
Copied!
from omnidocs.tasks.math_expression_extraction.extractors.unimernet import UniMERNetExtractor
print("UnimernetExtractor imported successfully!")
from omnidocs.tasks.math_expression_extraction.extractors.unimernet import UniMERNetExtractor
print("UnimernetExtractor imported successfully!")
In [ ]:
Copied!
image_path = "../../../../tests/math_expression_extraction/assets/math_equation.png"
extractor = UniMERNetExtractor(device='cuda', show_log=False)
result = extractor.extract(image_path)
print(f"Unimernet: Found {len(result.expressions)} expressions")
image_path = "../../../../tests/math_expression_extraction/assets/math_equation.png"
extractor = UniMERNetExtractor(device='cuda', show_log=False)
result = extractor.extract(image_path)
print(f"Unimernet: Found {len(result.expressions)} expressions")