复现基于GoogleNet的图像分类中遇到的问题
1.配置docker开发环境
参考:配置 Docker 开发环境 | Milk-V (milkv.io)
2.在Dokcer中准备工作目录
建立工作目录与tpu-mlir同级
mkdir googlenet && cd googlenet
wget https://github.com/onnx/models/raw/main/vision/classification/inception_and_goog
lenet/googlenet/model/googlenet-12.onnx
cp -rf $TPUC_ROOT/regression/dataset/ILSVRC2012 .
cp -rf $TPUC_ROOT/regression/image .
新建work目录
mkdir work && cd work
3.ONNX转MLIR
model_transform.py \
--model_name googlenet \
--model_def ../googlenet-12.onnx \
--test_input ../image/cat.jpg \
--input_shapes [[1,3,224,224]] \
--resize_dims 256,256 \
--mean 123.675,116.28,103.53 \
--scale 0.0171,0.0175,0.0174 \
--pixel_format rgb \
--test_result googlenet_top_outputs.npz \
--mlir googlenet.mlir
运行失败了
4.总结
ONNX转MLIR运行失败了 求解