關于将.pt檔案轉換成onx檔案

看到網上沒有關于.pt轉化成onx檔案的教程

首先打開anaconda prompt

建立一個虛拟環境叫做onx

話不多說直接上代碼

pip install onnx
           

安裝pyTorch

打開網站https://pytorch.org/get-started/locally/

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
           

接下來繼續安裝

pip install onnx-caffe2
           

pip install onnx-simplifier
           

然後直接cmd進入需要轉換的檔案夾

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

發現需要pandas，安裝pandas

pip install pandas
           

後面繼續報錯我就直接把需要的直接貼上來了

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

           

所有的都安裝完成後 

直接輸入上面的代碼

可以就看到結果

下面就直接放上轉化後的onx的結果圖像需要進一步分析的小夥伴可以下載下傳

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