20 lines
439 B
Python
20 lines
439 B
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from models.ocr_model.utils.inference import inference as latex_inference
|
||
|
|
|
||
|
|
|
||
|
|
def load_lang_models(language: str):
|
||
|
|
...
|
||
|
|
# language: 'ch' or 'en'
|
||
|
|
# return det_model, rec_model (or model)
|
||
|
|
|
||
|
|
|
||
|
|
def load_det_tex_model():
|
||
|
|
...
|
||
|
|
# return the loaded latex detection model
|
||
|
|
|
||
|
|
|
||
|
|
def mix_inference(latex_det_model, latex_rec_model, lang_model, img: np.ndarray, use_cuda: bool) -> str:
|
||
|
|
...
|
||
|
|
# latex_inference(...)
|