Skip to content

Commit 38fe6ea

Browse files
committed
Update Sphinx documentation
1 parent d86a301 commit 38fe6ea

16 files changed

Lines changed: 467 additions & 103 deletions
Binary file not shown.
Binary file not shown.
Binary file not shown.
13.6 KB
Binary file not shown.

docs/py_docs/build/html/_modules/perceptionmetrics/datasets/coco.html

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,8 @@
239239
</div>
240240
<article role="main" id="furo-main-content">
241241
<h1>Source code for perceptionmetrics.datasets.coco</h1><div class="highlight"><pre>
242-
<span></span><span class="kn">from</span><span class="w"> </span><span class="nn">pycocotools.coco</span><span class="w"> </span><span class="kn">import</span> <span class="n">COCO</span>
242+
<span></span><span class="kn">from</span><span class="w"> </span><span class="nn">typing</span><span class="w"> </span><span class="kn">import</span> <span class="n">OrderedDict</span>
243+
<span class="kn">from</span><span class="w"> </span><span class="nn">pycocotools.coco</span><span class="w"> </span><span class="kn">import</span> <span class="n">COCO</span>
243244
<span class="kn">import</span><span class="w"> </span><span class="nn">os</span>
244245
<span class="kn">import</span><span class="w"> </span><span class="nn">re</span>
245246
<span class="kn">import</span><span class="w"> </span><span class="nn">pandas</span><span class="w"> </span><span class="k">as</span><span class="w"> </span><span class="nn">pd</span>
@@ -334,18 +335,18 @@ <h1>Source code for perceptionmetrics.datasets.coco</h1><div class="highlight"><
334335
<span class="p">}</span>
335336

336337
<span class="c1"># Build dataset DataFrame from COCO image IDs</span>
337-
<span class="n">rows</span> <span class="o">=</span> <span class="p">[]</span>
338+
<span class="n">dataset</span> <span class="o">=</span> <span class="n">OrderedDict</span><span class="p">()</span>
338339
<span class="k">for</span> <span class="n">img_id</span> <span class="ow">in</span> <span class="n">coco</span><span class="o">.</span><span class="n">getImgIds</span><span class="p">():</span>
339340
<span class="n">img_info</span> <span class="o">=</span> <span class="n">coco</span><span class="o">.</span><span class="n">loadImgs</span><span class="p">(</span><span class="n">img_id</span><span class="p">)[</span><span class="mi">0</span><span class="p">]</span>
340-
<span class="n">rows</span><span class="o">.</span><span class="n">append</span><span class="p">(</span>
341-
<span class="p">{</span>
342-
<span class="s2">&quot;image&quot;</span><span class="p">:</span> <span class="n">img_info</span><span class="p">[</span><span class="s2">&quot;file_name&quot;</span><span class="p">],</span>
343-
<span class="s2">&quot;annotation&quot;</span><span class="p">:</span> <span class="nb">str</span><span class="p">(</span><span class="n">img_id</span><span class="p">),</span>
344-
<span class="s2">&quot;split&quot;</span><span class="p">:</span> <span class="n">split</span><span class="p">,</span> <span class="c1"># Use provided split parameter</span>
345-
<span class="p">}</span>
341+
<span class="n">sample_name</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">basename</span><span class="p">(</span><span class="n">img_info</span><span class="p">[</span><span class="s2">&quot;file_name&quot;</span><span class="p">])</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s2">&quot;.&quot;</span><span class="p">)[</span><span class="mi">0</span><span class="p">]</span>
342+
<span class="n">dataset</span><span class="p">[</span><span class="n">sample_name</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
343+
<span class="n">img_info</span><span class="p">[</span><span class="s2">&quot;file_name&quot;</span><span class="p">],</span>
344+
<span class="nb">str</span><span class="p">(</span><span class="n">img_id</span><span class="p">),</span>
345+
<span class="n">split</span><span class="p">,</span> <span class="c1"># Use provided split parameter</span>
346346
<span class="p">)</span>
347347

348-
<span class="n">dataset</span> <span class="o">=</span> <span class="n">pd</span><span class="o">.</span><span class="n">DataFrame</span><span class="p">(</span><span class="n">rows</span><span class="p">)</span>
348+
<span class="n">cols</span> <span class="o">=</span> <span class="p">[</span><span class="s2">&quot;image&quot;</span><span class="p">,</span> <span class="s2">&quot;annotation&quot;</span><span class="p">,</span> <span class="s2">&quot;split&quot;</span><span class="p">]</span>
349+
<span class="n">dataset</span> <span class="o">=</span> <span class="n">pd</span><span class="o">.</span><span class="n">DataFrame</span><span class="o">.</span><span class="n">from_dict</span><span class="p">(</span><span class="n">dataset</span><span class="p">,</span> <span class="n">orient</span><span class="o">=</span><span class="s2">&quot;index&quot;</span><span class="p">,</span> <span class="n">columns</span><span class="o">=</span><span class="n">cols</span><span class="p">)</span>
349350
<span class="n">dataset</span><span class="o">.</span><span class="n">attrs</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;ontology&quot;</span><span class="p">:</span> <span class="n">ontology</span><span class="p">}</span>
350351

351352
<span class="k">return</span> <span class="n">dataset</span><span class="p">,</span> <span class="n">ontology</span></div>

0 commit comments

Comments
 (0)