YOLOX导出onnx文件，Python语言基础教程

export_onnx.py

#!/usr/bin/env python3

-- coding:utf-8 --

Copyright © Megvii, Inc. and its affiliates.

import argparse

import os

from loguru import logger

import torch

from torch import nn

from yolox.exp import get_exp

from yolox.models.network_blocks import SiLU

from yolox.utils import replace_module

def make_parser():

parser = argparse.ArgumentParser(“YOLOX onnx deploy”)

parser.add_argument(

“–output-name”, type=str, default=“yolox.onnx”, help=“output name of models”

)

parser.add_argument(

“–input”, default=“images”, type=str, help=“input node name of onnx model”

)

parser.add_argument(

“–output”, default=“output”, type=str, help=“output node name of onnx model”

)

pa