关于将.pt文件转换成onx文件

看到网上没有关于.pt转化成onx文件的教程

首先打开anaconda prompt

创建一个虚拟环境叫做onx

话不多说直接上代码

pip install onnx

关于将.pt文件转换成onx文件_第1张图片

安装pyTorch

打开网站https://pytorch.org/get-started/locally/

关于将.pt文件转换成onx文件_第2张图片

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

接下来继续安装

pip install onnx-caffe2

pip install onnx-simplifier

关于将.pt文件转换成onx文件_第3张图片

然后直接cmd进入需要转换的文件夹

python3 models / export.py --weights models / yolov5s.pt --img 640 --batch 1

关于将.pt文件转换成onx文件_第4张图片

发现需要pandas,安装pandas

pip install pandas

关于将.pt文件转换成onx文件_第5张图片

后面继续报错我就直接把需要的直接贴上来了

ModuleNotFoundError: No module named 'requests'

ModuleNotFoundError: No module named 'cv2'

ModuleNotFoundError: No module named 'tqdm'

ModuleNotFoundError: No module named 'yaml'

ModuleNotFoundError: No module named 'matplotlib'

ModuleNotFoundError: No module named 'seaborn'

pip install requests

pip install opencv-python

pip install tqdm

pip install pyyaml

pip install matplotlib

pip install seaborn

所有的都安装完成后 

直接输入上面的代码

关于将.pt文件转换成onx文件_第6张图片

可以就看到结果

下面就直接放上转化后的onx的结果图像需要进一步分析的小伙伴可以下载

界面https://download.csdn.net/user/qwazp3526cn/uploads?spm=3001.5449

你可能感兴趣的:(计算机视觉,神经网络,深度学习,pytorch,caffe)