Yuantian Gao
Results
1
comments of
Yuantian Gao
mmdeploy转换配置即 classification_onnxruntime_static.py 推理代码即为官方代码 其中配置文件为上述的mmpretrain训练生成的配置文件, 推理代码如下 `from mmdeploy.apis.utils import build_task_processor from mmdeploy.utils import get_input_shape, load_config import torch deploy_cfg = 'configs/mmpretrain/classification_onnxruntime_dynamic.py' model_cfg = './resnet18_8xb32_in1k.py' device = 'cpu' backend_model = ['./mmdeploy_models/mmpretrain/ort/end2end.onnx'] image =...