1、背景
在《标注工具Labelimg的安装与使用》中说明了如何安装及创建一个虚拟环境。但这个虚拟环境是如何使用的,特别是与VSCode如何配合使用的,进行说明和记录
2、说明
1、创建了一个虚拟环境
bash
conda create --name=onnxdemo python=3.10
2、运行如上的python环境
bash
conda activate onnxdemo
3、在环境中安装ultralytics
安装ultralytics进行yolo模型的导出(将导出为onnx格式)
bash
pip install ultralytics
这样,在指定的环境中就安装相关的package
4、打开VSCode,快捷键方式
bash
Ctrl+Shift+P
5、在输入框中输入:pyhton...

选中第一项
6、选择我们在conda中设置的环境

7、VScode的状态显示
VScode的右下角就有我们的环境显示

8、在编辑器中新建python文件
python
from ultralytics import YOLO
# 加载一个预训练的 YOLO 模型,例如 yolov8n.pt
model = YOLO("yolov8n.pt")
# 导出为 ONNX 格式
success = model.export(format="onnx", opset=17)
9、运行程序。

这样就能得到一个onnx格式的模型文件