From a4bb77d38d7b2dbccfadb9783caf7e1860786865 Mon Sep 17 00:00:00 2001 From: xujiao Date: Mon, 1 Dec 2025 11:19:15 +0800 Subject: [PATCH] =?UTF-8?q?test:=E6=9B=B4=E6=96=B0=E9=9B=86=E6=88=90?= =?UTF-8?q?=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/integration_test/test_caffe_conversion.py | 10 +++++----- test/integration_test/test_darknet_conversion.py | 8 ++++---- test/integration_test/test_onnx_conversion.py | 8 ++++---- test/integration_test/test_pytorch_conversion.py | 8 ++++---- test/integration_test/test_tensorflow_conversion.py | 8 ++++---- 5 files changed, 21 insertions(+), 21 deletions(-) diff --git a/test/integration_test/test_caffe_conversion.py b/test/integration_test/test_caffe_conversion.py index abe5fd9..b1a4c9b 100644 --- a/test/integration_test/test_caffe_conversion.py +++ b/test/integration_test/test_caffe_conversion.py @@ -7,11 +7,11 @@ import os import pytest from pathlib import Path from netrans import Netrans -from quantize_types import QuantizerType +from netrans.quantize_types import QuantizerType from shutil import copytree, rmtree import tempfile -ROOT = Path(__file__).parent.parent +ROOT = Path(__file__).parent.parent.parent # 通用转换函数 def _convert(model_dir: Path): @@ -21,14 +21,14 @@ def _convert(model_dir: Path): model_dir (Path): _description_ """ model = Netrans() - qtypes = list(QuantizerType.get_options()) + qtypes = list(QuantizerType.get_options())[:4] # qtypes = [list(QuantizerType.get_options())[14]] # qtypes= ['Ai16Wi4', 'Ai16Wpcqi8', 'Ai16Wpcqi4', 'Adfpi16Wpcqi8', 'Adfpi16Wpcqi4']#, 'Au10Wpcqi8', 'Au16Wi8', 'Au16Wpcqi8', 'Afp16Wpcqi8', 'Afp16Wpcqi4'] for qtype in qtypes: model.load(model_path=str(model_dir)) model.quantize(quantized=qtype) - model.add_prepost_to_graph(quantized=qtype) - model.export_nbg(quantized=qtype) + model.add_pre_post(quantized=qtype) + model.export(quantized=qtype) # 创建临时目录并执行测试 def _test_conversion(model_dir: Path, test_func): diff --git a/test/integration_test/test_darknet_conversion.py b/test/integration_test/test_darknet_conversion.py index 5803918..e2ca6f7 100644 --- a/test/integration_test/test_darknet_conversion.py +++ b/test/integration_test/test_darknet_conversion.py @@ -7,11 +7,11 @@ import os import pytest from pathlib import Path from netrans import Netrans -from quantize_types import QuantizerType +from netrans.quantize_types import QuantizerType from shutil import copytree, rmtree import tempfile -ROOT = Path(__file__).parent.parent +ROOT = Path(__file__).parent.parent.parent # 通用转换函数 def _convert(model_dir: Path): @@ -25,8 +25,8 @@ def _convert(model_dir: Path): for qtype in qtypes: model.load(model_path=str(model_dir), mean=[0,0,0], scale=[0.0039216,0.0039216,0.0039216]) model.quantize(quantized=qtype) - model.add_prepost_to_graph(quantized=qtype) - model.export_nbg(quantized=qtype) + model.add_pre_post(quantized=qtype) + model.export(quantized=qtype) # 创建临时目录并执行测试 def _test_conversion(model_dir: Path, test_func): diff --git a/test/integration_test/test_onnx_conversion.py b/test/integration_test/test_onnx_conversion.py index 9ba103b..ef3f2c0 100644 --- a/test/integration_test/test_onnx_conversion.py +++ b/test/integration_test/test_onnx_conversion.py @@ -7,11 +7,11 @@ import os import pytest from pathlib import Path from netrans import Netrans -from quantize_types import QuantizerType +from netrans.quantize_types import QuantizerType from shutil import copytree, rmtree import tempfile -ROOT = Path(__file__).parent.parent +ROOT = Path(__file__).parent.parent.parent # 通用转换函数 def _convert(model_dir: Path): @@ -25,8 +25,8 @@ def _convert(model_dir: Path): for qtype in qtypes: model.load(model_path=str(model_dir), mean=0, scale=[0.0039216]*3) model.quantize(quantized=qtype) - model.add_prepost_to_graph(quantized=qtype) - model.export_nbg(quantized=qtype) + model.add_pre_post(quantized=qtype) + model.export(quantized=qtype) # 创建临时目录并执行测试 def _test_conversion(model_dir: Path, test_func): diff --git a/test/integration_test/test_pytorch_conversion.py b/test/integration_test/test_pytorch_conversion.py index ef934bb..cfde854 100644 --- a/test/integration_test/test_pytorch_conversion.py +++ b/test/integration_test/test_pytorch_conversion.py @@ -7,11 +7,11 @@ import os import pytest from pathlib import Path from netrans import Netrans -from quantize_types import QuantizerType +from netrans.quantize_types import QuantizerType from shutil import copytree, rmtree import tempfile -ROOT = Path(__file__).parent.parent +ROOT = Path(__file__).parent.parent.parent # 通用转换函数 def _convert(model_dir: Path): @@ -25,8 +25,8 @@ def _convert(model_dir: Path): for qtype in qtypes: model.load(model_path=str(model_dir), mean=[0,0,0], scale=[0.0039216,0.0039216,0.0039216]) model.quantize(quantized=qtype) - model.add_prepost_to_graph(quantized=qtype) - model.export_nbg(quantized=qtype) + model.add_pre_post(quantized=qtype) + model.export(quantized=qtype) # 创建临时目录并执行测试 def _test_conversion(model_dir: Path, test_func): diff --git a/test/integration_test/test_tensorflow_conversion.py b/test/integration_test/test_tensorflow_conversion.py index bf8c9ee..2aafa19 100644 --- a/test/integration_test/test_tensorflow_conversion.py +++ b/test/integration_test/test_tensorflow_conversion.py @@ -7,11 +7,11 @@ import os import pytest from pathlib import Path from netrans import Netrans -from quantize_types import QuantizerType +from netrans.quantize_types import QuantizerType from shutil import copytree, rmtree import tempfile -ROOT = Path(__file__).parent.parent +ROOT = Path(__file__).parent.parent.parent # 通用转换函数 def _convert(model_dir: Path): @@ -25,8 +25,8 @@ def _convert(model_dir: Path): for qtype in qtypes: model.load(model_path=str(model_dir), mean=[0], scale=[0.0039216]) model.quantize(quantized=qtype) - model.add_prepost_to_graph(quantized=qtype) - model.export_nbg(quantized=qtype) + model.add_pre_post(quantized=qtype) + model.export(quantized=qtype) # 创建临时目录并执行测试 def _test_conversion(model_dir: Path, test_func): -- 2.34.1