26 lines
1.0 KiB
Python
26 lines
1.0 KiB
Python
|
# export_1280.py
|
|||
|
|
from ultralytics import YOLO
|
|||
|
|
|
|||
|
|
# 加载你的模型
|
|||
|
|
from ultralytics import YOLO
|
|||
|
|
|
|||
|
|
# 加载一个模型，路径为 YOLO 模型的 .pt 文件
|
|||
|
|
model = YOLO(r"/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_obb/weights/best.pt")
|
|||
|
|
|
|||
|
|
# 导出模型，设置多种参数
|
|||
|
|
model.export(
|
|||
|
|
format="onnx", # 导出格式为 ONNX
|
|||
|
|
imgsz=(640, 640), # 设置输入图像的尺寸
|
|||
|
|
keras=False, # 不导出为 Keras 格式
|
|||
|
|
optimize=False, # 不进行优化 False, 移动设备优化的参数，用于在导出为TorchScript 格式时进行模型优化
|
|||
|
|
half=False, # 不启用 FP16 量化
|
|||
|
|
int8=False, # 不启用 INT8 量化
|
|||
|
|
dynamic=False, # 不启用动态输入尺寸
|
|||
|
|
simplify=True, # 简化 ONNX 模型
|
|||
|
|
opset=None, # 使用最新的 opset 版本
|
|||
|
|
workspace=4.0, # 为 TensorRT 优化设置最大工作区大小（GiB）
|
|||
|
|
nms=False, # 不添加 NMS（非极大值抑制）
|
|||
|
|
batch=1, # 指定批处理大小
|
|||
|
|
device="cpu" # 指定导出设备为CPU或GPU，对应参数为"cpu" , "0"
|
|||
|
|
)
|