源码解析二 模型转换 export.py

技术源码解析二 模型转换 export.py 源码解析二 模型转换 export.py基于最新yolov5-v6.01.重点2.相关函数1.重点一般使用规则:python export.py --dat

源码解析二模型转换export.py

基于最新yolov5-v6.0

1.重点

2.相关函数

1.重点

一般使用规则:

python导出。py-数据' '-重量' '-img SZ 512-简化-包括' onnx '三种格式想要用哪种就要下载相应的包:

torchscript不需要下载对应的包有火炬就可以onnx: pip安装onnxcoreml: pip安装coremltools2 .相关函数

parse_opt():

def parse_opt():

'''

数据:数据集目录默认=ROOT/'data/coco128.yaml '

重量:权重文件目录默认=ROOT/'yolov5s.pt '

img-size:输入模型的图片尺寸=(高度、宽度)默认=[640, 640]

批次大小:批次大小默认=1

设备:模型运行设备库达设备,即0或0,1,2,3或中央处理器默认中央处理器

包括:要将元素铂的符号文件转为什么格式可以为单个原始也可以为目录默认=['torchscript ',' onnx ',' coreml']

half:是否使用半精度FP16export转换默认=假

位置:是否设置YOLOv5就地检测()=真默认=假

列车:是否开启model.train()模式默认=真coreml转换必须为真实的

优化: TorchScript转化参数是否进行移动端优化默认=假

int8:支持CoreML/TF INT8量化不支持ONNX

dynamic: ONNX转换参数动态轴ONNX转换是否要进行批处理变量默认=假

simplify: ONNX转换参数是否简化onnx模型默认=假

opset: ONNX转换参数设置ONNX版本默认=13

topk-per-class: TF.js每一类别都要保留默认=100

topk-all: TF.js Topk为所有班级保留

IOu-thres 3360 TF。js IOu阈值默认值=0.45

conf-thres3360 TF。射流研究…置信度阈值默认值=0.25

包括:需要导出的版本默认=['torchscript ',' onnx'],

' def run(data=ROOT/' data/coco 128。yaml ',# '数据集. YAML路径'

def run(data=ROOT/' data/coco 128。yaml ',# '数据集. YAML路径'

权重=ROOT/'yolov5s.pt ',#权重路径

imgsz=(640,640),# image(高度,宽度)

批处理大小=1,#批处理大小

设备='cpu ',# cuda设备,即0或0,1,2,3或中央处理器

包括=(“火炬脚本”、“onnx”、“coreml”)、#包括格式

一半=假,# FP16半精度导出

在位=假,#设置YOLOv5检测()在位=真

train=False,# model.train()模式

优化=假,# TorchScript:针对移动优化

int8=False,# CoreML/TF INT8量化

动态=假,# ONNX/TF:动态轴

简化=假,# ONNX:简化模型

opset=12,# ONNX: opset版本

topk_per_class=100,# TF.js NMS: topk每类保留

topk_all=100,# TF.js NMS: topk为所有类保留

iou_thres=0.45,# TF.js NMS: IoU阈值

conf_thres=0.25 # TF.js NMS:置信阈值

):

内容来源网络,如有侵权,联系删除,本文地址:https://www.230890.com/zhan/154010.html

(0)

相关推荐

  • SequoiaDB 分布式事务实现原理是什么

    技术SequoiaDB 分布式事务实现原理是什么这篇文章将为大家详细讲解有关SequoiaDB 分布式事务实现原理是什么,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。1分

    攻略 2021年11月23日
  • 4行4字的古诗,我要60首4行5字的古诗

    技术4行4字的古诗,我要60首4行5字的古诗1,静夜思4行4字的古诗。2,咏鹅。3,终南望余雪。4,登鹳雀楼。5,江雪。6,寻隐者不遇。7,江上渔者。8,夏日绝句。9,所见。10,春夜喜雨。11,八阵图。12,绝句。13

    生活 2021年10月23日
  • 描写梨花的优美古诗,几句赞美梨花的优美语句

    技术描写梨花的优美古诗,几句赞美梨花的优美语句1、清晨,白璧无瑕,大地从沉睡中清醒,梨花也充满着蓬勃的朝气描写梨花的优美古诗。梨花没有玫瑰的妖娆美丽,没有菊花的傲然脱俗,然而它却是我最喜爱的花,雪白的花瓣在露珠的清洗下变

    生活 2021年10月29日
  • 太极拳入门基本功,初学太极拳最好的方法有哪几种

    技术太极拳入门基本功,初学太极拳最好的方法有哪几种初学太极拳最好的方法有哪几种太极拳入门基本功?(原创)这个问题我来回答题主。我从五十岁开始学太极拳,到现在己经有二十三年了。根据我学习太极拳的经历来回答这个问题。初学大极

    生活 2021年10月22日
  • 本周最新文献速递20211114

    技术本周最新文献速递20211114 本周最新文献速递20211114本周最新文献速递20211114一、精细解读文献 一
    文献题目: Mapping the proteo-genomic conver

    礼包 2021年11月15日
  • C++面试易错语法特性有哪些

    技术C++面试易错语法特性有哪些这篇文章主要讲解了“C++面试易错语法特性有哪些”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“C++面试易错语法特性有哪些”吧!面试频率高

    攻略 2021年11月29日