mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix unit tests (#300)

pull/304/head
AllentDan 2021-06-18 18:27:20 +08:00 committed by GitHub
parent f1b003ddb1
commit a86ab115e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/test_core/test_deploy_utils.py
View File

@ -17,13 +17,13 @@ from mmocr.core.deployment import (ONNXRuntimeDetector, ONNXRuntimeRecognizer,
reason='skip if torch=1.3.x')
@pytest.mark.skipif(
not torch.cuda.is_available(), reason='skip if on cpu device')
@pytest.mark.importorskip('onnxruntime')
@pytest.mark.importorskip('tensorrt')
@pytest.mark.importorskip('mmcv.tensorrt')
def test_detector_wraper():
import onnxruntime as ort # noqa: F401
import tensorrt as trt
from mmcv.tensorrt import (onnx2trt, save_trt_engine)
try:
import onnxruntime as ort # noqa: F401
import tensorrt as trt
from mmcv.tensorrt import (onnx2trt, save_trt_engine)
except ImportError:
pytest.skip('ONNXRuntime or TensorRT is not available.')
onnx_path = 'tmp.onnx'
cfg = dict(
@ -121,13 +121,13 @@ def test_detector_wraper():
reason='skip if torch=1.3.x')
@pytest.mark.skipif(
not torch.cuda.is_available(), reason='skip if on cpu device')
@pytest.mark.importorskip('onnxruntime')
@pytest.mark.importorskip('tensorrt')
@pytest.mark.importorskip('mmcv.tensorrt')
def test_recognizer_wraper():
import onnxruntime as ort # noqa: F401
import tensorrt as trt
from mmcv.tensorrt import (onnx2trt, save_trt_engine)
try:
import onnxruntime as ort # noqa: F401
import tensorrt as trt
from mmcv.tensorrt import (onnx2trt, save_trt_engine)
except ImportError:
pytest.skip('ONNXRuntime or TensorRT is not available.')
onnx_path = 'tmp.onnx'
cfg = dict(