Skip to content

Commit 588cd2c

Browse files
committed
Update Sphinx documentation
1 parent 98d4b73 commit 588cd2c

2 files changed

Lines changed: 7 additions & 3 deletions

File tree

185 Bytes
Binary file not shown.

docs/py_docs/build/html/_modules/perceptionmetrics/utils/detection_metrics.html

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -376,9 +376,13 @@ <h1>Source code for perceptionmetrics.utils.detection_metrics</h1><div class="hi
376376

377377
<span class="n">ious</span> <span class="o">=</span> <span class="n">compute_iou_matrix</span><span class="p">(</span><span class="n">pred_boxes</span><span class="p">,</span> <span class="n">gt_boxes</span><span class="p">)</span> <span class="c1"># shape: (num_preds, num_gts)</span>
378378

379-
<span class="k">for</span> <span class="n">i</span><span class="p">,</span> <span class="p">(</span><span class="n">p_box</span><span class="p">,</span> <span class="n">p_label</span><span class="p">,</span> <span class="n">score</span><span class="p">)</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span>
380-
<span class="nb">zip</span><span class="p">(</span><span class="n">pred_boxes</span><span class="p">,</span> <span class="n">pred_labels</span><span class="p">,</span> <span class="n">pred_scores</span><span class="p">)</span>
381-
<span class="p">):</span>
379+
<span class="c1"># Sort predictions by descending score to ensure highest-confidence predictions match first</span>
380+
<span class="n">sorted_indices</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">argsort</span><span class="p">([</span><span class="o">-</span><span class="n">s</span> <span class="k">for</span> <span class="n">s</span> <span class="ow">in</span> <span class="n">pred_scores</span><span class="p">])</span>
381+
382+
<span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="n">sorted_indices</span><span class="p">:</span>
383+
<span class="n">p_box</span> <span class="o">=</span> <span class="n">pred_boxes</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
384+
<span class="n">p_label</span> <span class="o">=</span> <span class="n">pred_labels</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
385+
<span class="n">score</span> <span class="o">=</span> <span class="n">pred_scores</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
382386
<span class="n">max_iou</span> <span class="o">=</span> <span class="mi">0</span>
383387
<span class="n">max_j</span> <span class="o">=</span> <span class="o">-</span><span class="mi">1</span>
384388

0 commit comments

Comments
 (0)