From 5301e97afc16d328a91dfbe5c6d80f4d41ab5e11 Mon Sep 17 00:00:00 2001 From: inisis Date: Fri, 1 Nov 2024 23:53:30 +0800 Subject: [PATCH] update benchmark --- tests/test_benchmark.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py index 38c13a0..ad0abbc 100644 --- a/tests/test_benchmark.py +++ b/tests/test_benchmark.py @@ -104,6 +104,16 @@ def test_uiex(self, request): filename = f"{MODELZOO_PATH}/{name}/{name}.onnx" self.run_model_test(name, filename) + def test_en_number_mobile_v2_0_rec_infer(self, request): + name = request.node.originalname[len("test_") :] + filename = f"{MODELZOO_PATH}/{name}/{name}.onnx" + self.run_model_test(name, filename) + + def test_paddleocr(self, request): + name = request.node.originalname[len("test_") :] + filename = f"{MODELZOO_PATH}/{name}/{name}.onnx" + self.run_model_test(name, filename) + if __name__ == "__main__": pytest.main(