pip install tpu_mlir[onnx]
在docker中,新建文件夹model_yolov5s用于存放相关文件以及进行转换操作,并进入文件夹
mkdir model_yolov5s && cd model_yolov5s
将图片和模型复制到model_yolov5s文件夹中:
cp -rf tpu_mlir_resource/dataset/COCO2017 .
如果这里和我一样报错的话,需要先退出model_yolov5s文件夹,命令为: cd …
然后进行复制:
cp -rf yolov5s.onnx ./model_yolov5s
cp -rf tpu_mlir_resource/image ./model_yolov5s
cp -rf tpu_mlir_resource/dataset/COCO2017 ./model_yolov5s
注意:这里也可以先新建model_yolov5s文件夹,直接从本地使用docker cp命令到 model