forked from nudt_dsp/netrans
267 lines
15 KiB
HTML
267 lines
15 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" /><meta name="viewport" content="width=device-width, initial-scale=1" />
|
|
|
|
<title>import_model module — 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="next" title="infer module" href="infer.html" />
|
|
<link rel="prev" title="file_model module" href="file_model.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">
|
|
|
|
<section id="module-import_model">
|
|
<span id="import-model-module"></span><h1>import_model module<a class="headerlink" href="#module-import_model" title="Link to this heading">¶</a></h1>
|
|
<dl class="py class">
|
|
<dt class="sig sig-object py" id="import_model.ImportModel">
|
|
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">ImportModel</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">source_obj</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#ImportModel"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.ImportModel" title="Link to this definition">¶</a></dt>
|
|
<dd><p>基类：<a class="reference internal" href="utils.html#utils.AttributeCopier" title="utils.AttributeCopier"><code class="xref py py-class docutils literal notranslate"><span class="pre">AttributeCopier</span></code></a></p>
|
|
<p>从实例化的 Netrans 中解析模型参数,并基于 pnnacc 导入模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p><strong>Netrans</strong> (<em>class</em>) -- 实例化的Netrans类,包含 模型信息 和 Netrans 信息</p>
|
|
</dd>
|
|
</dl>
|
|
<dl class="py method">
|
|
<dt class="sig sig-object py" id="import_model.ImportModel.import_network">
|
|
<span class="sig-name descname"><span class="pre">import_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kargs</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#import_model.ImportModel.import_network" title="Link to this definition">¶</a></dt>
|
|
<dd></dd></dl>
|
|
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.check_status">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">check_status</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">result</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#check_status"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.check_status" title="Link to this definition">¶</a></dt>
|
|
<dd><p>解析命令执行情况</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p><strong>result</strong> (<em>return</em><em> of </em><em>subprocrss.run</em>) -- subprocess.run的返回值</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.import_caffe_network">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">import_caffe_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">netrans_path</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#import_caffe_network"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.import_caffe_network" title="Link to this definition">¶</a></dt>
|
|
<dd><p>导入 caffe 模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><ul class="simple">
|
|
<li><p><strong>name</strong> (<em>str</em>) -- 模型名字</p></li>
|
|
<li><p><strong>netrans_path</strong> (<em>str</em>) -- 模型路径</p></li>
|
|
</ul>
|
|
</dd>
|
|
<dt class="field-even">返回<span class="colon">:</span></dt>
|
|
<dd class="field-even"><p>生成的pnnacc 命令行, 被subprocesses执行</p>
|
|
</dd>
|
|
<dt class="field-odd">返回类型<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p>cmd (str)</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.import_darknet_network">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">import_darknet_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">netrans_path</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#import_darknet_network"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.import_darknet_network" title="Link to this definition">¶</a></dt>
|
|
<dd><p>导入 darknet 模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><ul class="simple">
|
|
<li><p><strong>name</strong> (<em>str</em>) -- 模型名字</p></li>
|
|
<li><p><strong>netrans_path</strong> (<em>str</em>) -- 模型路径</p></li>
|
|
</ul>
|
|
</dd>
|
|
<dt class="field-even">返回<span class="colon">:</span></dt>
|
|
<dd class="field-even"><p>生成的pnnacc 命令行, 被subprocesses执行</p>
|
|
</dd>
|
|
<dt class="field-odd">返回类型<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p>cmd (str)</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.import_onnx_network">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">import_onnx_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">netrans_path</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#import_onnx_network"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.import_onnx_network" title="Link to this definition">¶</a></dt>
|
|
<dd><p>导入 onnx 模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><ul class="simple">
|
|
<li><p><strong>name</strong> (<em>str</em>) -- 模型名字</p></li>
|
|
<li><p><strong>netrans_path</strong> (<em>str</em>) -- 模型路径</p></li>
|
|
</ul>
|
|
</dd>
|
|
<dt class="field-even">返回<span class="colon">:</span></dt>
|
|
<dd class="field-even"><p>生成的pnnacc 命令行, 被subprocesses执行</p>
|
|
</dd>
|
|
<dt class="field-odd">返回类型<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p>cmd (str)</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.import_pytorch_network">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">import_pytorch_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">netrans_path</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#import_pytorch_network"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.import_pytorch_network" title="Link to this definition">¶</a></dt>
|
|
<dd><p>导入 pytorch 模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><ul class="simple">
|
|
<li><p><strong>name</strong> (<em>str</em>) -- 模型名字</p></li>
|
|
<li><p><strong>netrans_path</strong> (<em>str</em>) -- 模型路径</p></li>
|
|
</ul>
|
|
</dd>
|
|
<dt class="field-even">返回<span class="colon">:</span></dt>
|
|
<dd class="field-even"><p>生成的pnnacc 命令行, 被subprocesses执行</p>
|
|
</dd>
|
|
<dt class="field-odd">返回类型<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p>cmd (str)</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.import_tensorflow_network">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">import_tensorflow_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">netrans_path</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#import_tensorflow_network"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.import_tensorflow_network" title="Link to this definition">¶</a></dt>
|
|
<dd><p>导入 tensorflow 模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><ul class="simple">
|
|
<li><p><strong>name</strong> (<em>str</em>) -- 模型名字</p></li>
|
|
<li><p><strong>netrans_path</strong> (<em>str</em>) -- 模型路径</p></li>
|
|
</ul>
|
|
</dd>
|
|
<dt class="field-even">返回<span class="colon">:</span></dt>
|
|
<dd class="field-even"><p>生成的pnnacc 命令行, 被subprocesses执行</p>
|
|
</dd>
|
|
<dt class="field-odd">返回类型<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p>cmd (str)</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
<dl class="py function">
|
|
<dt class="sig sig-object py" id="import_model.import_tflite_network">
|
|
<span class="sig-prename descclassname"><span class="pre">import_model.</span></span><span class="sig-name descname"><span class="pre">import_tflite_network</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">netrans_path</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="_modules/import_model.html#import_tflite_network"><span class="viewcode-link"><span class="pre">[源代码]</span></span></a><a class="headerlink" href="#import_model.import_tflite_network" title="Link to this definition">¶</a></dt>
|
|
<dd><p>导入 tflite 模型</p>
|
|
<dl class="field-list simple">
|
|
<dt class="field-odd">参数<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><ul class="simple">
|
|
<li><p><strong>name</strong> (<em>str</em>) -- 模型名字</p></li>
|
|
<li><p><strong>netrans_path</strong> (<em>str</em>) -- 模型路径</p></li>
|
|
</ul>
|
|
</dd>
|
|
<dt class="field-even">返回<span class="colon">:</span></dt>
|
|
<dd class="field-even"><p>生成的pnnacc 命令行, 被subprocesses执行</p>
|
|
</dd>
|
|
<dt class="field-odd">返回类型<span class="colon">:</span></dt>
|
|
<dd class="field-odd"><p>cmd (str)</p>
|
|
</dd>
|
|
</dl>
|
|
</dd></dl>
|
|
|
|
</section>
|
|
|
|
|
|
</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 class="current">
|
|
<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 current"><a class="reference internal" href="appendix.html">附录</a><ul class="current">
|
|
<li class="toctree-l2"><a class="reference internal" href="gen_api.html">gen api html & pdf by sphinx</a></li>
|
|
<li class="toctree-l2 current"><a class="reference internal" href="modules.html">netrans_py</a></li>
|
|
</ul>
|
|
</li>
|
|
</ul>
|
|
|
|
<div class="relations">
|
|
<h3>Related Topics</h3>
|
|
<ul>
|
|
<li><a href="index.html">Documentation overview</a><ul>
|
|
<li><a href="appendix.html">附录</a><ul>
|
|
<li><a href="modules.html">netrans_py</a><ul>
|
|
<li>Previous: <a href="file_model.html" title="上一章">file_model module</a></li>
|
|
<li>Next: <a href="infer.html" title="下一章">infer module</a></li>
|
|
</ul></li>
|
|
</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>
|
|
|
|
|
|
|
<a href="_sources/import_model.rst.txt"
|
|
rel="nofollow">Page source</a>
|
|
</div>
|
|
|
|
|
|
|
|
|
|
</body>
|
|
</html> |