mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update docs

gh-pages
KaiyangZhou 2019-10-23 17:17:08 +01:00
parent 6ecc568a7c
commit 4df051b750
45 changed files with 15 additions and 153 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
_modules/torchreid/data/datasets/dataset.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">import</span> <span class="nn">tarfile</span>
@ -350,7 +348,6 @@
<span class="n">download_url</span><span class="p">(</span><span class="n">dataset_url</span><span class="p">,</span> <span class="n">fpath</span><span class="p">)</span>
<span class="nb">print</span><span class="p">(</span><span class="s1">&#39;Extracting &quot;</span><span class="si">{}</span><span class="s1">&quot;&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">fpath</span><span class="p">))</span>
<span class="n">extension</span> <span class="o">=</span> <span class="n">osp</span><span class="o">.</span><span class="n">basename</span><span class="p">(</span><span class="n">fpath</span><span class="p">)</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s1">&#39;.&#39;</span><span class="p">)[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
<span class="k">try</span><span class="p">:</span>
<span class="n">tar</span> <span class="o">=</span> <span class="n">tarfile</span><span class="o">.</span><span class="n">open</span><span class="p">(</span><span class="n">fpath</span><span class="p">)</span>
<span class="n">tar</span><span class="o">.</span><span class="n">extractall</span><span class="p">(</span><span class="n">path</span><span class="o">=</span><span class="n">dataset_dir</span><span class="p">)</span>

2
_modules/torchreid/data/datasets/image/cuhk01.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">zipfile</span>

2
_modules/torchreid/data/datasets/image/cuhk03.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">from</span> <span class="nn">torchreid.data.datasets</span> <span class="k">import</span> <span class="n">ImageDataset</span>

2
_modules/torchreid/data/datasets/image/dukemtmcreid.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">re</span>

2
_modules/torchreid/data/datasets/image/grid.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">from</span> <span class="nn">scipy.io</span> <span class="k">import</span> <span class="n">loadmat</span>

7
_modules/torchreid/data/datasets/image/ilids.html
View File

@ -169,11 +169,8 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">import</span> <span class="nn">copy</span>
<span class="kn">import</span> <span class="nn">random</span>
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">defaultdict</span>
@ -245,11 +242,10 @@
<span class="s1">&#39;but got </span><span class="si">{}</span><span class="s1">, please check the data&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">num_pids</span><span class="p">)</span>
<span class="n">num_train_pids</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">num_pids</span> <span class="o">*</span> <span class="mf">0.5</span><span class="p">)</span>
<span class="n">num_test_pids</span> <span class="o">=</span> <span class="n">num_pids</span> <span class="o">-</span> <span class="n">num_train_pids</span> <span class="c1"># supposed to be 60</span>
<span class="n">splits</span> <span class="o">=</span> <span class="p">[]</span>
<span class="k">for</span> <span class="n">_</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">10</span><span class="p">):</span>
<span class="c1"># randomly choose num_train_pids train IDs and num_test_pids test IDs</span>
<span class="c1"># randomly choose num_train_pids train IDs and the rest for test IDs</span>
<span class="n">pids_copy</span> <span class="o">=</span> <span class="n">copy</span><span class="o">.</span><span class="n">deepcopy</span><span class="p">(</span><span class="n">pids</span><span class="p">)</span>
<span class="n">random</span><span class="o">.</span><span class="n">shuffle</span><span class="p">(</span><span class="n">pids_copy</span><span class="p">)</span>
<span class="n">train_pids</span> <span class="o">=</span> <span class="n">pids_copy</span><span class="p">[:</span><span class="n">num_train_pids</span><span class="p">]</span>
@ -301,7 +297,6 @@
<span class="k">return</span> <span class="n">data</span>
<span class="k">def</span> <span class="nf">process_split</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">split</span><span class="p">):</span>
<span class="n">train</span><span class="p">,</span> <span class="n">query</span><span class="p">,</span> <span class="n">gallery</span> <span class="o">=</span> <span class="p">[],</span> <span class="p">[],</span> <span class="p">[]</span>
<span class="n">train_pid2label</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_pid2label</span><span class="p">(</span><span class="n">split</span><span class="p">[</span><span class="s1">&#39;train&#39;</span><span class="p">])</span>
<span class="n">train</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">parse_img_names</span><span class="p">(</span><span class="n">split</span><span class="p">[</span><span class="s1">&#39;train&#39;</span><span class="p">],</span> <span class="n">train_pid2label</span><span class="p">)</span>
<span class="n">query</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">parse_img_names</span><span class="p">(</span><span class="n">split</span><span class="p">[</span><span class="s1">&#39;query&#39;</span><span class="p">])</span>

2
_modules/torchreid/data/datasets/image/market1501.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">re</span>

2
_modules/torchreid/data/datasets/image/msmt17.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">from</span> <span class="nn">torchreid.data.datasets</span> <span class="k">import</span> <span class="n">ImageDataset</span>

3
_modules/torchreid/data/datasets/image/prid.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">random</span>
@ -242,7 +240,6 @@
<span class="nb">print</span><span class="p">(</span><span class="s1">&#39;Split file is saved to </span><span class="si">{}</span><span class="s1">&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">split_path</span><span class="p">))</span>
<span class="k">def</span> <span class="nf">process_split</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">split</span><span class="p">):</span>
<span class="n">train</span><span class="p">,</span> <span class="n">query</span><span class="p">,</span> <span class="n">gallery</span> <span class="o">=</span> <span class="p">[],</span> <span class="p">[],</span> <span class="p">[]</span>
<span class="n">train_pids</span> <span class="o">=</span> <span class="n">split</span><span class="p">[</span><span class="s1">&#39;train&#39;</span><span class="p">]</span>
<span class="n">test_pids</span> <span class="o">=</span> <span class="n">split</span><span class="p">[</span><span class="s1">&#39;test&#39;</span><span class="p">]</span>

2
_modules/torchreid/data/datasets/image/sensereid.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">copy</span>

2
_modules/torchreid/data/datasets/image/viper.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>

2
_modules/torchreid/data/datasets/video/dukemtmcvidreid.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">import</span> <span class="nn">warnings</span>

2
_modules/torchreid/data/datasets/video/ilidsvid.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">from</span> <span class="nn">scipy.io</span> <span class="k">import</span> <span class="n">loadmat</span>

6
_modules/torchreid/data/datasets/video/mars.html
View File

@ -169,8 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">from</span> <span class="nn">scipy.io</span> <span class="k">import</span> <span class="n">loadmat</span>
<span class="kn">import</span> <span class="nn">warnings</span>
@ -243,9 +241,9 @@
<span class="k">assert</span> <span class="n">home_dir</span> <span class="ow">in</span> <span class="p">[</span><span class="s1">&#39;bbox_train&#39;</span><span class="p">,</span> <span class="s1">&#39;bbox_test&#39;</span><span class="p">]</span>
<span class="n">num_tracklets</span> <span class="o">=</span> <span class="n">meta_data</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
<span class="n">pid_list</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">meta_data</span><span class="p">[:,</span><span class="mi">2</span><span class="p">]</span><span class="o">.</span><span class="n">tolist</span><span class="p">()))</span>
<span class="n">num_pids</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">pid_list</span><span class="p">)</span>
<span class="k">if</span> <span class="n">relabel</span><span class="p">:</span> <span class="n">pid2label</span> <span class="o">=</span> <span class="p">{</span><span class="n">pid</span><span class="p">:</span><span class="n">label</span> <span class="k">for</span> <span class="n">label</span><span class="p">,</span> <span class="n">pid</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">pid_list</span><span class="p">)}</span>
<span class="k">if</span> <span class="n">relabel</span><span class="p">:</span>
<span class="n">pid2label</span> <span class="o">=</span> <span class="p">{</span><span class="n">pid</span><span class="p">:</span><span class="n">label</span> <span class="k">for</span> <span class="n">label</span><span class="p">,</span> <span class="n">pid</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">pid_list</span><span class="p">)}</span>
<span class="n">tracklets</span> <span class="o">=</span> <span class="p">[]</span>
<span class="k">for</span> <span class="n">tracklet_idx</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">num_tracklets</span><span class="p">):</span>

4
_modules/torchreid/data/datasets/video/prid2011.html
View File

@ -169,13 +169,11 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">glob</span>
<span class="kn">from</span> <span class="nn">torchreid.data.datasets</span> <span class="k">import</span> <span class="n">VideoDataset</span>
<span class="kn">from</span> <span class="nn">torchreid.utils</span> <span class="k">import</span> <span class="n">read_json</span><span class="p">,</span> <span class="n">write_json</span>
<span class="kn">from</span> <span class="nn">torchreid.utils</span> <span class="k">import</span> <span class="n">read_json</span>
<div class="viewcode-block" id="PRID2011"><a class="viewcode-back" href="../../../../../pkg/data.html#torchreid.data.datasets.video.prid2011.PRID2011">[docs]</a><span class="k">class</span> <span class="nc">PRID2011</span><span class="p">(</span><span class="n">VideoDataset</span><span class="p">):</span>

1
_modules/torchreid/data/sampler.html
View File

@ -173,7 +173,6 @@
<span class="kn">import</span> <span class="nn">copy</span>
<span class="kn">import</span> <span class="nn">random</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch.utils.data.sampler</span> <span class="k">import</span> <span class="n">Sampler</span><span class="p">,</span> <span class="n">RandomSampler</span>

1
_modules/torchreid/data/transforms.html
View File

@ -171,7 +171,6 @@
<span class="kn">from</span> <span class="nn">PIL</span> <span class="k">import</span> <span class="n">Image</span>
<span class="kn">import</span> <span class="nn">random</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">import</span> <span class="nn">math</span>
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">deque</span>

6
_modules/torchreid/engine/engine.html
View File

@ -169,22 +169,17 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">datetime</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">import</span> <span class="nn">cv2</span>
<span class="kn">from</span> <span class="nn">matplotlib</span> <span class="k">import</span> <span class="n">pyplot</span> <span class="k">as</span> <span class="n">plt</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">from</span> <span class="nn">torch.utils.tensorboard</span> <span class="k">import</span> <span class="n">SummaryWriter</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<span class="kn">from</span> <span class="nn">torchreid.utils</span> <span class="k">import</span> <span class="n">AverageMeter</span><span class="p">,</span> <span class="n">visualize_ranked_results</span><span class="p">,</span> <span class="n">save_checkpoint</span><span class="p">,</span> <span class="n">re_ranking</span><span class="p">,</span> <span class="n">mkdir_if_missing</span>
<span class="kn">from</span> <span class="nn">torchreid.losses</span> <span class="k">import</span> <span class="n">DeepSupervision</span>
<span class="kn">from</span> <span class="nn">torchreid</span> <span class="k">import</span> <span class="n">metrics</span>
@ -403,7 +398,6 @@
<span class="nb">print</span><span class="p">(</span><span class="s1">&#39;Extracting features from gallery set ...&#39;</span><span class="p">)</span>
<span class="n">gf</span><span class="p">,</span> <span class="n">g_pids</span><span class="p">,</span> <span class="n">g_camids</span> <span class="o">=</span> <span class="p">[],</span> <span class="p">[],</span> <span class="p">[]</span> <span class="c1"># gallery features, gallery person IDs and gallery camera IDs</span>
<span class="n">end</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
<span class="k">for</span> <span class="n">batch_idx</span><span class="p">,</span> <span class="n">data</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">galleryloader</span><span class="p">):</span>
<span class="n">imgs</span><span class="p">,</span> <span class="n">pids</span><span class="p">,</span> <span class="n">camids</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_parse_data_for_eval</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">use_gpu</span><span class="p">:</span>

8
_modules/torchreid/engine/image/softmax.html
View File

@ -172,16 +172,13 @@
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">datetime</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<span class="kn">from</span> <span class="nn">torchreid.engine</span> <span class="k">import</span> <span class="n">engine</span>
<span class="kn">from</span> <span class="nn">torchreid.engine</span> <span class="k">import</span> <span class="n">Engine</span>
<span class="kn">from</span> <span class="nn">torchreid.losses</span> <span class="k">import</span> <span class="n">CrossEntropyLoss</span>
<span class="kn">from</span> <span class="nn">torchreid.utils</span> <span class="k">import</span> <span class="n">AverageMeter</span><span class="p">,</span> <span class="n">open_specified_layers</span><span class="p">,</span> <span class="n">open_all_layers</span>
<span class="kn">from</span> <span class="nn">torchreid</span> <span class="k">import</span> <span class="n">metrics</span>
<div class="viewcode-block" id="ImageSoftmaxEngine"><a class="viewcode-back" href="../../../../pkg/engine.html#torchreid.engine.image.softmax.ImageSoftmaxEngine">[docs]</a><span class="k">class</span> <span class="nc">ImageSoftmaxEngine</span><span class="p">(</span><span class="n">engine</span><span class="o">.</span><span class="n">Engine</span><span class="p">):</span>
<div class="viewcode-block" id="ImageSoftmaxEngine"><a class="viewcode-back" href="../../../../pkg/engine.html#torchreid.engine.image.softmax.ImageSoftmaxEngine">[docs]</a><span class="k">class</span> <span class="nc">ImageSoftmaxEngine</span><span class="p">(</span><span class="n">Engine</span><span class="p">):</span>
<span class="sa">r</span><span class="sd">&quot;&quot;&quot;Softmax-loss engine for image-reid.</span>
<span class="sd"> Args:</span>
@ -195,7 +192,6 @@
<span class="sd"> Examples::</span>
<span class="sd"> </span>
<span class="sd"> import torch</span>
<span class="sd"> import torchreid</span>
<span class="sd"> datamanager = torchreid.data.ImageDataManager(</span>
<span class="sd"> root=&#39;path/to/reid-data&#39;,</span>

8
_modules/torchreid/engine/image/triplet.html
View File

@ -172,16 +172,13 @@
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">datetime</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<span class="kn">from</span> <span class="nn">torchreid.engine</span> <span class="k">import</span> <span class="n">engine</span>
<span class="kn">from</span> <span class="nn">torchreid.engine</span> <span class="k">import</span> <span class="n">Engine</span>
<span class="kn">from</span> <span class="nn">torchreid.losses</span> <span class="k">import</span> <span class="n">CrossEntropyLoss</span><span class="p">,</span> <span class="n">TripletLoss</span>
<span class="kn">from</span> <span class="nn">torchreid.utils</span> <span class="k">import</span> <span class="n">AverageMeter</span><span class="p">,</span> <span class="n">open_specified_layers</span><span class="p">,</span> <span class="n">open_all_layers</span>
<span class="kn">from</span> <span class="nn">torchreid</span> <span class="k">import</span> <span class="n">metrics</span>
<div class="viewcode-block" id="ImageTripletEngine"><a class="viewcode-back" href="../../../../pkg/engine.html#torchreid.engine.image.triplet.ImageTripletEngine">[docs]</a><span class="k">class</span> <span class="nc">ImageTripletEngine</span><span class="p">(</span><span class="n">engine</span><span class="o">.</span><span class="n">Engine</span><span class="p">):</span>
<div class="viewcode-block" id="ImageTripletEngine"><a class="viewcode-back" href="../../../../pkg/engine.html#torchreid.engine.image.triplet.ImageTripletEngine">[docs]</a><span class="k">class</span> <span class="nc">ImageTripletEngine</span><span class="p">(</span><span class="n">Engine</span><span class="p">):</span>
<span class="sa">r</span><span class="sd">&quot;&quot;&quot;Triplet-loss engine for image-reid.</span>
<span class="sd"> Args:</span>
@ -198,7 +195,6 @@
<span class="sd"> Examples::</span>
<span class="sd"> </span>
<span class="sd"> import torch</span>
<span class="sd"> import torchreid</span>
<span class="sd"> datamanager = torchreid.data.ImageDataManager(</span>
<span class="sd"> root=&#39;path/to/reid-data&#39;,</span>

4
_modules/torchreid/engine/video/softmax.html
View File

@ -169,12 +169,8 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">datetime</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<span class="kn">from</span> <span class="nn">torchreid.engine.image</span> <span class="k">import</span> <span class="n">ImageSoftmaxEngine</span>

6
_modules/torchreid/engine/video/triplet.html
View File

@ -169,12 +169,6 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">print_function</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">datetime</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<span class="kn">from</span> <span class="nn">torchreid.engine.image</span> <span class="k">import</span> <span class="n">ImageTripletEngine</span>
<span class="kn">from</span> <span class="nn">torchreid.engine.video</span> <span class="k">import</span> <span class="n">VideoSoftmaxEngine</span>

2
_modules/torchreid/metrics/rank.html
View File

@ -170,9 +170,7 @@
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">import</span> <span class="nn">copy</span>
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">defaultdict</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="kn">import</span> <span class="nn">warnings</span>
<span class="k">try</span><span class="p">:</span>

2
_modules/torchreid/models/densenet.html
View File

@ -174,14 +174,12 @@
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;densenet121&#39;</span><span class="p">,</span> <span class="s1">&#39;densenet169&#39;</span><span class="p">,</span> <span class="s1">&#39;densenet201&#39;</span><span class="p">,</span> <span class="s1">&#39;densenet161&#39;</span><span class="p">,</span> <span class="s1">&#39;densenet121_fc512&#39;</span><span class="p">]</span>
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">OrderedDict</span>
<span class="kn">import</span> <span class="nn">math</span>
<span class="kn">import</span> <span class="nn">re</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">from</span> <span class="nn">torch.utils</span> <span class="k">import</span> <span class="n">model_zoo</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="n">model_urls</span> <span class="o">=</span> <span class="p">{</span>

1
_modules/torchreid/models/hacnn.html
View File

@ -173,7 +173,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="k">class</span> <span class="nc">ConvBlock</span><span class="p">(</span><span class="n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>

43
_modules/torchreid/models/inceptionresnetv2.html
View File

@ -165,22 +165,17 @@
<div itemprop="articleBody">
<h1>Source code for torchreid.models.inceptionresnetv2</h1><div class="highlight"><pre>
<span></span><span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">absolute_import</span>
<span></span><span class="sd">&quot;&quot;&quot;</span>
<span class="sd">Code imported from https://github.com/Cadene/pretrained-models.pytorch</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">absolute_import</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;inceptionresnetv2&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd">Code imported from https://github.com/Cadene/pretrained-models.pytorch</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="n">pretrained_settings</span> <span class="o">=</span> <span class="p">{</span>
@ -407,36 +402,6 @@
<span class="k">return</span> <span class="n">out</span>
<span class="k">def</span> <span class="nf">inceptionresnetv2</span><span class="p">(</span><span class="n">num_classes</span><span class="o">=</span><span class="mi">1000</span><span class="p">,</span> <span class="n">pretrained</span><span class="o">=</span><span class="s1">&#39;imagenet&#39;</span><span class="p">):</span>
<span class="sa">r</span><span class="sd">&quot;&quot;&quot;InceptionResNetV2 model architecture from the</span>
<span class="sd"> `&quot;InceptionV4, Inception-ResNet...&quot; &lt;https://arxiv.org/abs/1602.07261&gt;`_ paper.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">if</span> <span class="n">pretrained</span><span class="p">:</span>
<span class="n">settings</span> <span class="o">=</span> <span class="n">pretrained_settings</span><span class="p">[</span><span class="s1">&#39;inceptionresnetv2&#39;</span><span class="p">][</span><span class="n">pretrained</span><span class="p">]</span>
<span class="k">assert</span> <span class="n">num_classes</span> <span class="o">==</span> <span class="n">settings</span><span class="p">[</span><span class="s1">&#39;num_classes&#39;</span><span class="p">],</span> \
<span class="s1">&#39;num_classes should be </span><span class="si">{}</span><span class="s1">, but is </span><span class="si">{}</span><span class="s1">&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">settings</span><span class="p">[</span><span class="s1">&#39;num_classes&#39;</span><span class="p">],</span> <span class="n">num_classes</span><span class="p">)</span>
<span class="c1"># both &#39;imagenet&#39;&amp;&#39;imagenet+background&#39; are loaded from same parameters</span>
<span class="n">model</span> <span class="o">=</span> <span class="n">InceptionResNetV2</span><span class="p">(</span><span class="n">num_classes</span><span class="o">=</span><span class="mi">1001</span><span class="p">)</span>
<span class="n">model</span><span class="o">.</span><span class="n">load_state_dict</span><span class="p">(</span><span class="n">model_zoo</span><span class="o">.</span><span class="n">load_url</span><span class="p">(</span><span class="n">settings</span><span class="p">[</span><span class="s1">&#39;url&#39;</span><span class="p">]))</span>
<span class="k">if</span> <span class="n">pretrained</span> <span class="o">==</span> <span class="s1">&#39;imagenet&#39;</span><span class="p">:</span>
<span class="n">new_last_linear</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">1536</span><span class="p">,</span> <span class="mi">1000</span><span class="p">)</span>
<span class="n">new_last_linear</span><span class="o">.</span><span class="n">weight</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">last_linear</span><span class="o">.</span><span class="n">weight</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="mi">1</span><span class="p">:]</span>
<span class="n">new_last_linear</span><span class="o">.</span><span class="n">bias</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">last_linear</span><span class="o">.</span><span class="n">bias</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="mi">1</span><span class="p">:]</span>
<span class="n">model</span><span class="o">.</span><span class="n">last_linear</span> <span class="o">=</span> <span class="n">new_last_linear</span>
<span class="n">model</span><span class="o">.</span><span class="n">input_space</span> <span class="o">=</span> <span class="n">settings</span><span class="p">[</span><span class="s1">&#39;input_space&#39;</span><span class="p">]</span>
<span class="n">model</span><span class="o">.</span><span class="n">input_size</span> <span class="o">=</span> <span class="n">settings</span><span class="p">[</span><span class="s1">&#39;input_size&#39;</span><span class="p">]</span>
<span class="n">model</span><span class="o">.</span><span class="n">input_range</span> <span class="o">=</span> <span class="n">settings</span><span class="p">[</span><span class="s1">&#39;input_range&#39;</span><span class="p">]</span>
<span class="n">model</span><span class="o">.</span><span class="n">mean</span> <span class="o">=</span> <span class="n">settings</span><span class="p">[</span><span class="s1">&#39;mean&#39;</span><span class="p">]</span>
<span class="n">model</span><span class="o">.</span><span class="n">std</span> <span class="o">=</span> <span class="n">settings</span><span class="p">[</span><span class="s1">&#39;std&#39;</span><span class="p">]</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">model</span> <span class="o">=</span> <span class="n">InceptionResNetV2</span><span class="p">(</span><span class="n">num_classes</span><span class="o">=</span><span class="n">num_classes</span><span class="p">)</span>
<span class="k">return</span> <span class="n">model</span>
<span class="c1">##################### Model Definition #########################</span>

4
_modules/torchreid/models/inceptionv4.html
View File

@ -167,14 +167,12 @@
<h1>Source code for torchreid.models.inceptionv4</h1><div class="highlight"><pre>
<span></span><span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">absolute_import</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="k">import</span> <span class="n">division</span>
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;inceptionv4&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">sys</span>
<span class="sd">&quot;&quot;&quot;</span>

1
_modules/torchreid/models/mlfn.html
View File

@ -173,7 +173,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

1
_modules/torchreid/models/mobilenetv2.html
View File

@ -170,7 +170,6 @@
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;mobilenetv2_x1_0&#39;</span><span class="p">,</span> <span class="s1">&#39;mobilenetv2_x1_4&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

1
_modules/torchreid/models/mudeep.html
View File

@ -173,7 +173,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="k">class</span> <span class="nc">ConvBlock</span><span class="p">(</span><span class="n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>

1
_modules/torchreid/models/nasnet.html
View File

@ -174,7 +174,6 @@
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">import</span> <span class="nn">torch.nn.functional</span> <span class="k">as</span> <span class="nn">F</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="sd">&quot;&quot;&quot;</span>

1
_modules/torchreid/models/osnet.html
View File

@ -173,7 +173,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="n">pretrained_urls</span> <span class="o">=</span> <span class="p">{</span>

2
_modules/torchreid/models/pcb.html
View File

@ -170,10 +170,8 @@
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;pcb_p6&#39;</span><span class="p">,</span> <span class="s1">&#39;pcb_p4&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

3
_modules/torchreid/models/resnet.html
View File

@ -174,10 +174,7 @@
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;resnet18&#39;</span><span class="p">,</span> <span class="s1">&#39;resnet34&#39;</span><span class="p">,</span> <span class="s1">&#39;resnet50&#39;</span><span class="p">,</span> <span class="s1">&#39;resnet101&#39;</span><span class="p">,</span> <span class="s1">&#39;resnet152&#39;</span><span class="p">,</span> <span class="s1">&#39;resnext50_32x4d&#39;</span><span class="p">,</span>
<span class="s1">&#39;resnext101_32x8d&#39;</span><span class="p">,</span> <span class="s1">&#39;resnet50_fc512&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

2
_modules/torchreid/models/resnetmid.html
View File

@ -172,8 +172,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

3
_modules/torchreid/models/senet.html
View File

@ -181,11 +181,8 @@
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">OrderedDict</span>
<span class="kn">import</span> <span class="nn">math</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">from</span> <span class="nn">torch.utils</span> <span class="k">import</span> <span class="n">model_zoo</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="sd">&quot;&quot;&quot;</span>

1
_modules/torchreid/models/shufflenet.html
View File

@ -173,7 +173,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

2
_modules/torchreid/models/shufflenetv2.html
View File

@ -175,8 +175,6 @@
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="k">import</span> <span class="n">nn</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

6
_modules/torchreid/models/squeezenet.html
View File

@ -177,15 +177,9 @@
<span class="s1">&#39;squeezenet1_0_fc512&#39;</span>
<span class="p">]</span>
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">OrderedDict</span>
<span class="kn">import</span> <span class="nn">math</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">from</span> <span class="nn">torch.utils</span> <span class="k">import</span> <span class="n">model_zoo</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">functional</span> <span class="k">as</span> <span class="n">F</span>
<span class="kn">import</span> <span class="nn">torch.nn.init</span> <span class="k">as</span> <span class="nn">init</span>
<span class="kn">import</span> <span class="nn">torchvision</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>

3
_modules/torchreid/models/xception.html
View File

@ -170,12 +170,9 @@
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;xception&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">math</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">import</span> <span class="nn">torch.nn.functional</span> <span class="k">as</span> <span class="nn">F</span>
<span class="kn">import</span> <span class="nn">torch.utils.model_zoo</span> <span class="k">as</span> <span class="nn">model_zoo</span>
<span class="kn">from</span> <span class="nn">torch.nn</span> <span class="k">import</span> <span class="n">init</span>
<span class="n">pretrained_settings</span> <span class="o">=</span> <span class="p">{</span>

1
_modules/torchreid/utils/model_complexity.html
View File

@ -175,7 +175,6 @@
<span class="kn">from</span> <span class="nn">itertools</span> <span class="k">import</span> <span class="n">repeat</span>
<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="sd">&quot;&quot;&quot;</span>

2
_modules/torchreid/utils/reidtools.html
View File

@ -171,11 +171,9 @@
<span class="n">__all__</span> <span class="o">=</span> <span class="p">[</span><span class="s1">&#39;visualize_ranked_results&#39;</span><span class="p">]</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">import</span> <span class="nn">shutil</span>
<span class="kn">import</span> <span class="nn">cv2</span>
<span class="kn">from</span> <span class="nn">matplotlib</span> <span class="k">import</span> <span class="n">pyplot</span> <span class="k">as</span> <span class="n">plt</span>
<span class="kn">from</span> <span class="nn">.tools</span> <span class="k">import</span> <span class="n">mkdir_if_missing</span>

2
_modules/torchreid/utils/tools.html
View File

@ -178,7 +178,6 @@
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">errno</span>
<span class="kn">import</span> <span class="nn">json</span>
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">OrderedDict</span>
<span class="kn">import</span> <span class="nn">warnings</span>
<span class="kn">import</span> <span class="nn">random</span>
<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
@ -280,7 +279,6 @@
<span class="n">got_img</span> <span class="o">=</span> <span class="kc">True</span>
<span class="k">except</span> <span class="ne">IOError</span><span class="p">:</span>
<span class="nb">print</span><span class="p">(</span><span class="s1">&#39;IOError incurred when reading &quot;</span><span class="si">{}</span><span class="s1">&quot;. Will redo. Don</span><span class="se">\&#39;</span><span class="s1">t worry. Just chill.&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">path</span><span class="p">))</span>
<span class="k">pass</span>
<span class="k">return</span> <span class="n">img</span></div>

1
_modules/torchreid/utils/torchtools.html
View File

@ -176,7 +176,6 @@
<span class="kn">from</span> <span class="nn">collections</span> <span class="k">import</span> <span class="n">OrderedDict</span>
<span class="kn">import</span> <span class="nn">shutil</span>
<span class="kn">import</span> <span class="nn">warnings</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">os.path</span> <span class="k">as</span> <span class="nn">osp</span>
<span class="kn">from</span> <span class="nn">functools</span> <span class="k">import</span> <span class="n">partial</span>
<span class="kn">import</span> <span class="nn">pickle</span>

6
pkg/engine.html
View File

@ -315,8 +315,7 @@ or <code class="docutils literal notranslate"><span class="pre">torchreid.data.V
</tbody>
</table>
<p>Examples:</p>
<div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span> <span class="nn">torchreid</span>
<span class="n">datamanager</span> <span class="o">=</span> <span class="n">torchreid</span><span class="o">.</span><span class="n">data</span><span class="o">.</span><span class="n">ImageDataManager</span><span class="p">(</span>
<span class="n">root</span><span class="o">=</span><span class="s1">&#39;path/to/reid-data&#39;</span><span class="p">,</span>
<span class="n">sources</span><span class="o">=</span><span class="s1">&#39;market1501&#39;</span><span class="p">,</span>
@ -391,8 +390,7 @@ or <code class="docutils literal notranslate"><span class="pre">torchreid.data.V
</tbody>
</table>
<p>Examples:</p>
<div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">import</span> <span class="nn">torchreid</span>
<div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span> <span class="nn">torchreid</span>
<span class="n">datamanager</span> <span class="o">=</span> <span class="n">torchreid</span><span class="o">.</span><span class="n">data</span><span class="o">.</span><span class="n">ImageDataManager</span><span class="p">(</span>
<span class="n">root</span><span class="o">=</span><span class="s1">&#39;path/to/reid-data&#39;</span><span class="p">,</span>
<span class="n">sources</span><span class="o">=</span><span class="s1">&#39;market1501&#39;</span><span class="p">,</span>