YOLOX Deployment: ONNX export and ONNXRuntime Demo

释放双眼,带上耳机,听听看~!
Learn how to deploy YOLOX using ONNX export and ONNXRuntime. This article provides a step-by-step guide for deploying YOLOX models and testing the inference using ONNXRuntime.
  • 🥇 版权: 本文由【墨理学AI】认真原创、各位大佬、一文读懂、敬请查阅
  • 🎉 声明: 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
  • 👋 只有实战、不讲理论、一文读懂👋
  • 🥇 YOLOX Deployment

    📔 ONNX export and an ONNXRuntime

    所参考教程

    📕 环境搭建

    环境搭建,可以参考上篇博文

    🟧 export_onnx

    
    cd yoloDir
    
    git clone https://github.com/Megvii-BaseDetection/YOLOX.git
    
    cd YOLOX
    
    # 运行命令:
    
    python3 tools/export_onnx.py --output-name yolox_s.onnx -n yolox-s -c preModels/yolox_s.pth
    
    ## 或者,两种指定方式都可以
    
    # 如果说是,自己定义扩展的 YOLO ,那么 需要使用 -f 指定 xx.py 文件的方式
    
    python3 tools/export_onnx.py --output-name yolox_s.onnx -f exps/default/yolox_s.py -c preModels/yolox_s.pth
    

    输出如下【得到 yolox_s.onnx 】:

    ...
    
    2021-08-25 09:08:00.847 | INFO     | __main__:main:55 - args value: 
    
    Namespace(ckpt='preModels/yolox_s.pth', exp_file=None, experiment_name=None, input='images', name='yolox-s', no_onnxsim=False, opset=11, opts=[], output='output', output_name='yolox_s.onnx')
    2021-08-25 09:08:01.075 | INFO     | __main__:main:79 - loading checkpoint done.
    
    ## 先转为 yolox_s.onnx,然后 simplified 
    
    2021-08-25 09:08:06.185 | INFO     | __main__:main:89 - generated onnx model named yolox_s.onnx
    2021-08-25 09:08:08.043 | INFO     | __main__:main:101 - generated simplified onnx model named 
    
    yolox_s.onnx
    
    

    查看代码可以看到,–no-onnxsim 默认为 False

    YOLOX Deployment: ONNX export and ONNXRuntime Demo

    参数解析

    YOLOX Deployment: ONNX export and ONNXRuntime Demo

    🟨 ONNX Runtime Demo 【yolox_s.onnx 推理测试】

    
    cd YOLOX/demo/ONNXRuntime
    
    python3 onnx_inference.py -m ../../yolox_s.onnx -i ../../assets/dog.jpg -o ./ -s 0.3 --input_shape 640,640
    
    

    生成 和 原图 名字一致的 带检测图片,效果如下

    YOLOX Deployment: ONNX export and ONNXRuntime Demo

    参数解析

    YOLOX Deployment: ONNX export and ONNXRuntime Demo

  • ❤️ 你只看一篇就够用的教程 感谢各位大佬一键三连
  • YOLOX Deployment: ONNX export and ONNXRuntime Demo

    🚀🚀 墨理学AI

  • 🎉 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
  • ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作
  • YOLOX Deployment: ONNX export and ONNXRuntime Demo

    本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
    AI教程

    如何让 ChatGPT 输出高质量内容?学习提示词工程的有效技巧

    2023-12-14 14:52:00

    AI教程

    Meta AI推出的LLama 2:全面解读初学者指南

    2023-12-14 14:55:00

    个人中心
    购物车
    优惠劵
    今日签到
    有新私信 私信列表
    搜索