update readme

gh-pages
KaiyangZhou 2019-08-23 22:50:43 +01:00
parent e1610fee6a
commit c9c9d413dc
2 changed files with 20 additions and 4 deletions

View File

@ -296,9 +296,11 @@ $ conda activate torchreid
</div>
<div class="section" id="a-unified-interface">
<h2>A unified interface<a class="headerlink" href="#a-unified-interface" title="Permalink to this headline"></a></h2>
<p>In “deep-person-reid/scripts/”, we provide a unified interface including a default parser file <code class="docutils literal notranslate"><span class="pre">default_parser.py</span></code> and the main script <code class="docutils literal notranslate"><span class="pre">main.py</span></code>. For example, to train an image reid model on Market1501 using softmax, you can do</p>
<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>python main.py <span class="se">\</span>
--root path/to/reid-data <span class="se">\</span>
<p>In “deep-person-reid/scripts/”, we provide a unified interface to train and test a model.</p>
<p>For instance, to train an image reid model on Market1501 using softmax, you can do</p>
<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span><span class="c1"># suppose you are in deep-person-reid/</span>
python scripts/main.py <span class="se">\</span>
--root PATH_TO_DATA <span class="se">\</span>
--app image <span class="se">\</span>
--loss softmax <span class="se">\</span>
--label-smooth <span class="se">\</span>
@ -314,6 +316,20 @@ $ conda activate torchreid
--gpu-devices <span class="m">0</span>
</pre></div>
</div>
<p>To evaluate a trained model, do</p>
<div class="highlight-bash notranslate"><div class="highlight"><pre><span></span>python scripts/main.py <span class="se">\</span>
--root PATH_TO_DATA <span class="se">\</span>
--app image <span class="se">\</span>
--loss softmax <span class="se">\</span>
-s market1501 <span class="se">\</span>
-a resnet50 <span class="se">\</span>
--batch-size <span class="m">32</span> <span class="se">\</span>
--evaluate <span class="se">\</span>
--load-weights log/resnet50-market1501-softmax/model.pth.tar-60 <span class="se">\</span>
--save-dir log/eval-resnet50 <span class="se">\</span>
--gpu-devices <span class="m">0</span>
</pre></div>
</div>
<p>Please refer to <code class="docutils literal notranslate"><span class="pre">default_parser.py</span></code> and <code class="docutils literal notranslate"><span class="pre">main.py</span></code> for more details.</p>
</div>
<div class="section" id="datasets">

File diff suppressed because one or more lines are too long