mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-06-16 09:59:13 +02:00
28 lines
719 B
Python
28 lines
719 B
Python
from eynollah.model_zoo import EynollahModelZoo
|
|
from eynollah.predictor import Predictor
|
|
|
|
def test_trocr1(
|
|
model_dir,
|
|
):
|
|
model_zoo = EynollahModelZoo(model_dir)
|
|
try:
|
|
model_zoo.load_models(('ocr', 'tr'))
|
|
model = model_zoo.get('ocr')
|
|
assert isinstance(model, Predictor)
|
|
shape = model.input_shape
|
|
assert len(shape) == 3
|
|
except ImportError:
|
|
pass
|
|
|
|
def test_cnnrnnocr1(
|
|
model_dir,
|
|
):
|
|
model_zoo = EynollahModelZoo(model_dir)
|
|
try:
|
|
model_zoo.load_models('ocr')
|
|
model = model_zoo.get('ocr')
|
|
assert isinstance(model, Predictor)
|
|
shape = model.input_shape
|
|
assert len(shape) == 4
|
|
except ImportError:
|
|
pass
|