dfs</span></code> ，方便用户灵活选择。</p></li>
</ul>
</section>
<section id="id3">
<h2>模型定义及注册<a class="headerlink" href="#id3" title="Link to this heading">#</a></h2>
<p>接下来我们实现具体模型，在task下面新建models目录，并添加yolo.py用于YOLO模型定义，具体定义如下：</p>
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span c