yolov8模型转换

"""
Export a YOLOv8 PyTorch model to other formats. TensorFlow exports authored by https://github.com/zldrobit

Format                  | `format=argument`         | Model
---                     | ---                       | ---
PyTorch                 | -                         | yolov8n.pt
TorchScript             | `torchscript`             | yolov8n.torchscript
ONNX                    | `onnx`                    | yolov8n.onnx
OpenVINO                | `openvino`                | yolov8n_openvino_model/
TensorRT                | `engine`                  | yolov8n.engine
CoreML                  | `coreml`                  | yolov8n.mlmodel
TensorFlow SavedModel   | `saved_model`             | yolov8n_saved_model/
TensorFlow GraphDef     | `pb`                      | yolov8n.pb
TensorFlow Lite         | `tflite`                  | yolov8n.tflite
TensorFlow Edge TPU     | `edgetpu`                 | yolov8n_edgetpu.tflite
TensorFlow.js           | `tfjs`                    | yolov8n_web_model/
PaddlePaddle            | `paddle`                  | yolov8n_paddle_model/

Requirements:
    $ pip install ultralytics[export]

Python:
    from ultralytics import YOLO
    model = YOLO('yolov8n.pt')
    results = model.export(format='onnx')

CLI:
    $ yolo mode=export model=yolov8n.pt format=onnx

Inference:
    $ yolo predict model=yolov8n.pt                 # PyTorch
                         yolov8n.torchscript        # TorchScript
                         yolov8n.onnx               # ONNX Runtime or OpenCV DNN with --dnn
                         yolov8n_openvino_model     # OpenVINO
                         yolov8n.engine             # TensorRT
                         yolov8n.mlmodel            # CoreML (macOS-only)
                         yolov8n_saved_model        # TensorFlow SavedModel
                         yolov8n.pb                 # TensorFlow GraphDef
                         yolov8n.tflite             # TensorFlow Lite
                         yolov8n_edgetpu.tflite     # TensorFlow Edge TPU
                         yolov8n_paddle_model       # PaddlePaddle

TensorFlow.js:
    $ cd .. && git clone https://github.com/zldrobit/tfjs-yolov5-example.git && cd tfjs-yolov5-example
    $ npm install
    $ ln -s ../../yolov5/yolov8n_web_model public/yolov8n_web_model
    $ npm start
"""

相关推荐

  1. yolov8模型转换

    2024-01-12 17:34:02       48 阅读
  2. YOLOv8模型部署

    2024-01-12 17:34:02       57 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-12 17:34:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-12 17:34:02       100 阅读
  3. 在Django里面运行非项目文件

    2024-01-12 17:34:02       82 阅读
  4. Python语言-面向对象

    2024-01-12 17:34:02       91 阅读

热门阅读

  1. GNSS数据及产品下载地址(FTP/HTTP)

    2024-01-12 17:34:02       52 阅读
  2. SpringCloud + Redis 实现Api接口限流 防止恶意刷接口

    2024-01-12 17:34:02       50 阅读
  3. zip压缩工具类

    2024-01-12 17:34:02       48 阅读
  4. 龙芯系统部署Elasticsearch

    2024-01-12 17:34:02       51 阅读
  5. redis.conf 默认出厂内容

    2024-01-12 17:34:02       38 阅读
  6. 基于Redis的对象状态数据管理模块-后端开发

    2024-01-12 17:34:02       56 阅读
  7. kafka topic分区数设定

    2024-01-12 17:34:02       61 阅读
  8. 面试题目1

    2024-01-12 17:34:02       51 阅读