|
239 | 239 | </div> |
240 | 240 | <article role="main" id="furo-main-content"> |
241 | 241 | <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> |
243 | 244 | <span class="kn">import</span><span class="w"> </span><span class="nn">os</span> |
244 | 245 | <span class="kn">import</span><span class="w"> </span><span class="nn">re</span> |
245 | 246 | <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">< |
334 | 335 | <span class="p">}</span> |
335 | 336 |
|
336 | 337 | <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> |
338 | 339 | <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> |
339 | 340 | <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">"image"</span><span class="p">:</span> <span class="n">img_info</span><span class="p">[</span><span class="s2">"file_name"</span><span class="p">],</span> |
343 | | - <span class="s2">"annotation"</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">"split"</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">"file_name"</span><span class="p">])</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s2">"."</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">"file_name"</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> |
346 | 346 | <span class="p">)</span> |
347 | 347 |
|
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">"image"</span><span class="p">,</span> <span class="s2">"annotation"</span><span class="p">,</span> <span class="s2">"split"</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">"index"</span><span class="p">,</span> <span class="n">columns</span><span class="o">=</span><span class="n">cols</span><span class="p">)</span> |
349 | 350 | <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">"ontology"</span><span class="p">:</span> <span class="n">ontology</span><span class="p">}</span> |
350 | 351 |
|
351 | 352 | <span class="k">return</span> <span class="n">dataset</span><span class="p">,</span> <span class="n">ontology</span></div> |
|
0 commit comments