forked from nudt_dsp/netrans
173 lines
10 KiB
HTML
173 lines
10 KiB
HTML
<!DOCTYPE html>
|
||
|
||
<html lang="zh" data-content_root="../">
|
||
<head>
|
||
<meta charset="utf-8" />
|
||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||
<title>example — netrans 0.1 文档</title>
|
||
<link rel="stylesheet" type="text/css" href="../_static/pygments.css?v=5ecbeea2" />
|
||
<link rel="stylesheet" type="text/css" href="../_static/basic.css?v=b08954a9" />
|
||
<link rel="stylesheet" type="text/css" href="../_static/alabaster.css?v=27fed22d" />
|
||
<script src="../_static/documentation_options.js?v=52efc512"></script>
|
||
<script src="../_static/doctools.js?v=9bcbadda"></script>
|
||
<script src="../_static/sphinx_highlight.js?v=dc90522c"></script>
|
||
<link rel="index" title="索引" href="../genindex.html" />
|
||
<link rel="search" title="搜索" href="../search.html" />
|
||
|
||
<link rel="stylesheet" href="../_static/custom.css" type="text/css" />
|
||
|
||
|
||
|
||
|
||
|
||
</head><body>
|
||
|
||
|
||
<div class="document">
|
||
<div class="documentwrapper">
|
||
<div class="bodywrapper">
|
||
|
||
|
||
<div class="body" role="main">
|
||
|
||
<h1>example 源代码</h1><div class="highlight"><pre>
|
||
<span></span><span class="ch">#!/usr/bin/env python3</span>
|
||
|
||
<span class="kn">import</span><span class="w"> </span><span class="nn">argparse</span>
|
||
<span class="kn">from</span><span class="w"> </span><span class="nn">netrans</span><span class="w"> </span><span class="kn">import</span> <span class="n">Netrans</span>
|
||
|
||
<div class="viewcode-block" id="main">
|
||
<a class="viewcode-back" href="../example.html#example.main">[文档]</a>
|
||
<span class="k">def</span><span class="w"> </span><span class="nf">main</span><span class="p">():</span>
|
||
<span class="c1"># 创建参数解析器</span>
|
||
<span class="n">parser</span> <span class="o">=</span> <span class="n">argparse</span><span class="o">.</span><span class="n">ArgumentParser</span><span class="p">(</span>
|
||
<span class="n">description</span><span class="o">=</span><span class="s1">'神经网络模型转换工具'</span><span class="p">,</span>
|
||
<span class="n">formatter_class</span><span class="o">=</span><span class="n">argparse</span><span class="o">.</span><span class="n">ArgumentDefaultsHelpFormatter</span> <span class="c1"># 自动显示默认值</span>
|
||
<span class="p">)</span>
|
||
|
||
<span class="c1"># 必填位置参数</span>
|
||
<span class="n">parser</span><span class="o">.</span><span class="n">add_argument</span><span class="p">(</span>
|
||
<span class="s1">'model_path'</span><span class="p">,</span>
|
||
<span class="nb">type</span><span class="o">=</span><span class="nb">str</span><span class="p">,</span>
|
||
<span class="n">help</span><span class="o">=</span><span class="s1">'输入模型路径（必须参数）'</span>
|
||
<span class="p">)</span>
|
||
|
||
<span class="c1"># 可选参数组</span>
|
||
<span class="n">quant_group</span> <span class="o">=</span> <span class="n">parser</span><span class="o">.</span><span class="n">add_argument_group</span><span class="p">(</span><span class="s1">'量化参数'</span><span class="p">)</span>
|
||
<span class="n">quant_group</span><span class="o">.</span><span class="n">add_argument</span><span class="p">(</span>
|
||
<span class="s1">'-q'</span><span class="p">,</span> <span class="s1">'--quantize_type'</span><span class="p">,</span>
|
||
<span class="nb">type</span><span class="o">=</span><span class="nb">str</span><span class="p">,</span>
|
||
<span class="n">choices</span><span class="o">=</span><span class="p">[</span><span class="s1">'uint8'</span><span class="p">,</span> <span class="s1">'int8'</span><span class="p">,</span> <span class="s1">'int16'</span><span class="p">,</span> <span class="s1">'float'</span><span class="p">],</span>
|
||
<span class="n">default</span><span class="o">=</span><span class="s1">'uint8'</span><span class="p">,</span>
|
||
<span class="n">metavar</span><span class="o">=</span><span class="s1">'TYPE'</span><span class="p">,</span>
|
||
<span class="n">help</span><span class="o">=</span><span class="s1">'量化类型（可选值：</span><span class="si">%(choices)s</span><span class="s1">）'</span>
|
||
<span class="p">)</span>
|
||
<span class="n">quant_group</span><span class="o">.</span><span class="n">add_argument</span><span class="p">(</span>
|
||
<span class="s1">'-m'</span><span class="p">,</span> <span class="s1">'--mean'</span><span class="p">,</span>
|
||
<span class="nb">type</span><span class="o">=</span><span class="nb">int</span><span class="p">,</span>
|
||
<span class="n">default</span><span class="o">=</span><span class="mi">0</span><span class="p">,</span>
|
||
<span class="n">help</span><span class="o">=</span><span class="s1">'归一化均值（默认：</span><span class="si">%(default)s</span><span class="s1">）'</span>
|
||
<span class="p">)</span>
|
||
<span class="n">quant_group</span><span class="o">.</span><span class="n">add_argument</span><span class="p">(</span>
|
||
<span class="s1">'-s'</span><span class="p">,</span> <span class="s1">'--scale'</span><span class="p">,</span>
|
||
<span class="nb">type</span><span class="o">=</span><span class="nb">float</span><span class="p">,</span>
|
||
<span class="n">default</span><span class="o">=</span><span class="mf">1.0</span><span class="p">,</span>
|
||
<span class="n">help</span><span class="o">=</span><span class="s1">'量化缩放系数（默认：</span><span class="si">%(default)s</span><span class="s1">）'</span>
|
||
<span class="p">)</span>
|
||
<span class="n">parser</span><span class="o">.</span><span class="n">add_argument</span><span class="p">(</span>
|
||
<span class="s1">'-p'</span><span class="p">,</span> <span class="s1">'--profile'</span><span class="p">,</span>
|
||
<span class="n">action</span><span class="o">=</span><span class="s1">'store_true'</span><span class="p">,</span> <span class="c1"># 设置为True当参数存在时</span>
|
||
<span class="n">help</span><span class="o">=</span><span class="s1">'启用性能分析模式（默认：</span><span class="si">%(default)s</span><span class="s1">）'</span>
|
||
<span class="p">)</span>
|
||
|
||
|
||
<span class="c1"># 解析参数</span>
|
||
<span class="n">args</span> <span class="o">=</span> <span class="n">parser</span><span class="o">.</span><span class="n">parse_args</span><span class="p">()</span>
|
||
|
||
<span class="c1"># 执行模型转换</span>
|
||
<span class="k">try</span><span class="p">:</span>
|
||
<span class="n">model</span> <span class="o">=</span> <span class="n">Netrans</span><span class="p">(</span><span class="n">model_path</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">model_path</span><span class="p">)</span>
|
||
<span class="n">model</span><span class="o">.</span><span class="n">model2nbg</span><span class="p">(</span>
|
||
<span class="n">quantize_type</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">quantize_type</span><span class="p">,</span>
|
||
<span class="n">mean</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">mean</span><span class="p">,</span>
|
||
<span class="n">scale</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">scale</span><span class="p">,</span>
|
||
<span class="n">profile</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">profile</span>
|
||
<span class="p">)</span>
|
||
<span class="nb">print</span><span class="p">(</span><span class="sa">f</span><span class="s2">"模型 </span><span class="si">{</span><span class="n">args</span><span class="o">.</span><span class="n">model_path</span><span class="si">}</span><span class="s2"> 转换成功"</span><span class="p">)</span>
|
||
<span class="k">except</span> <span class="ne">FileNotFoundError</span><span class="p">:</span>
|
||
<span class="nb">print</span><span class="p">(</span><span class="sa">f</span><span class="s2">"错误：模型文件 </span><span class="si">{</span><span class="n">args</span><span class="o">.</span><span class="n">model_path</span><span class="si">}</span><span class="s2"> 不存在"</span><span class="p">)</span>
|
||
<span class="n">exit</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span></div>
|
||
|
||
|
||
<span class="k">if</span> <span class="vm">__name__</span> <span class="o">==</span> <span class="s2">"__main__"</span><span class="p">:</span>
|
||
<span class="n">main</span><span class="p">()</span>
|
||
</pre></div>
|
||
|
||
</div>
|
||
|
||
</div>
|
||
</div>
|
||
<div class="sphinxsidebar" role="navigation" aria-label="Main">
|
||
<div class="sphinxsidebarwrapper">
|
||
<h1 class="logo"><a href="../index.html">netrans</a></h1>
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
<search id="searchbox" style="display: none" role="search">
|
||
<div class="searchformwrapper">
|
||
<form class="search" action="../search.html" method="get">
|
||
<input type="text" name="q" aria-labelledby="searchlabel" autocomplete="off" autocorrect="off" autocapitalize="off" spellcheck="false" placeholder="Search"/>
|
||
<input type="submit" value="提交" />
|
||
</form>
|
||
</div>
|
||
</search>
|
||
<script>document.getElementById('searchbox').style.display = "block"</script><h3>导航</h3>
|
||
<p class="caption" role="heading"><span class="caption-text">Contents:</span></p>
|
||
<ul>
|
||
<li class="toctree-l1"><a class="reference internal" href="../quick_start_guide.html">快速入门</a></li>
|
||
<li class="toctree-l1"><a class="reference internal" href="../netrans_cli.html">netrans_cli 使用</a></li>
|
||
<li class="toctree-l1"><a class="reference internal" href="../netrans_py.html">netrans_py 使用</a></li>
|
||
<li class="toctree-l1"><a class="reference internal" href="../appendix.html">附录</a></li>
|
||
</ul>
|
||
|
||
<div class="relations">
|
||
<h3>Related Topics</h3>
|
||
<ul>
|
||
<li><a href="../index.html">Documentation overview</a><ul>
|
||
<li><a href="index.html">模块代码</a><ul>
|
||
</ul></li>
|
||
</ul></li>
|
||
</ul>
|
||
</div>
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
</div>
|
||
</div>
|
||
<div class="clearer"></div>
|
||
</div>
|
||
<div class="footer">
|
||
©2025, ccyh.
|
||
|
||
|
|
||
Powered by <a href="https://www.sphinx-doc.org/">Sphinx 8.2.3</a>
|
||
& <a href="https://alabaster.readthedocs.io">Alabaster 1.0.0</a>
|
||
|
||
</div>
|
||
|
||
|
||
|
||
|
||
</body>
|
||
</html> |