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”

)

parser.add_argument(

“-o”, “–opset”, default=11, type=int, help=“onnx opset version”

)

parser.add_argument(“–batch-size”, type=int, default=1, help=“batch size”)

parser.add_argument(

“–dynamic”, action=“store_true”, help=“whether the input shape should be dynamic or not”

)

parser.add_argument(“–no-onnxsim”, action=“store_true”, help=“use onnxsim or not”)

parser.add_argument(

“-f”,

“–exp_file”,

default=None,

type=str,

help=“expriment description file”,

)

parser.add_argument(“-expn”, “–experiment-name”, type=str, default=None)

parser.add_argument(“-n”, “–name”, type=str, default=None, help=“model name”)

parser.add_argument(“-c”, “–ckpt”, default=None, type=str, help=“ckpt path”)

parser.add_argument(

“opts”,

help=“Modify config options using the command-line”,

default=None,

nargs=argparse.REMAINDER,

)

return parser

@logger.catch

def main():

args = make_parser().parse_args()

logger.info(“args value: {}”.format(args))

exp = get_exp(args.exp_file, args.name)

exp.merge(args.opts)

if not args.experiment_name:

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python必备开发工具

工具都帮大家整理好了，安装就可直接上手！

三、最新Python学习笔记

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

四、Python视频合集

观看全面零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

五、实战案例

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

六、面试宝典

简历模板

小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数初中级Python工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Python爬虫全套学习资料》送给大家，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频

如果你觉得这些内容对你有帮助，可以添加下面V无偿领取！（备注：python）

不知道该从何学起的朋友，同时减轻大家的负担。**

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频

如果你觉得这些内容对你有帮助，可以添加下面V无偿领取！（备注：python）
[外链图片转存中…(img-Na7vTypk-1710881462169)]