Skip to content

Commit e74d2aa

Browse files
committed
Update Sphinx documentation
1 parent cdc5763 commit e74d2aa

4 files changed

Lines changed: 32 additions & 1 deletion

File tree

Binary file not shown.
1.32 KB
Binary file not shown.

docs/py_docs/build/html/_modules/perceptionmetrics/models/torch_segmentation.html

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -989,6 +989,8 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
989989
<span class="n">translation_direction</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;dataset_to_model&quot;</span><span class="p">,</span>
990990
<span class="n">predictions_outdir</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
991991
<span class="n">results_per_sample</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
992+
<span class="n">progress_callback</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Callable</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
993+
<span class="n">metrics_callback</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Callable</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
992994
<span class="p">)</span> <span class="o">-&gt;</span> <span class="n">pd</span><span class="o">.</span><span class="n">DataFrame</span><span class="p">:</span>
993995
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Perform evaluation for a LiDAR segmentation dataset</span>
994996

@@ -1004,6 +1006,10 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
10041006
<span class="sd"> :type predictions_outdir: Optional[str], optional</span>
10051007
<span class="sd"> :param results_per_sample: Whether to store results per sample or not, defaults to False. If True, predictions_outdir must be provided.</span>
10061008
<span class="sd"> :type results_per_sample: bool, optional</span>
1009+
<span class="sd"> :param progress_callback: Optional callback called as progress_callback(processed, total), defaults to None</span>
1010+
<span class="sd"> :type progress_callback: Optional[Callable], optional</span>
1011+
<span class="sd"> :param metrics_callback: Optional callback called as metrics_callback(metrics_df, processed, total), defaults to None</span>
1012+
<span class="sd"> :type metrics_callback: Optional[Callable], optional</span>
10071013
<span class="sd"> :return: DataFrame containing evaluation results</span>
10081014
<span class="sd"> :rtype: pd.DataFrame</span>
10091015
<span class="sd"> &quot;&quot;&quot;</span>
@@ -1053,6 +1059,9 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
10531059

10541060
<span class="c1"># Init metrics</span>
10551061
<span class="n">metrics_factory</span> <span class="o">=</span> <span class="n">um</span><span class="o">.</span><span class="n">SegmentationMetricsFactory</span><span class="p">(</span><span class="n">n_classes</span><span class="p">)</span>
1062+
<span class="n">total_samples</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">dataset</span><span class="p">)</span>
1063+
<span class="n">processed_samples</span> <span class="o">=</span> <span class="mi">0</span>
1064+
<span class="n">evaluation_step</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">model_cfg</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;evaluation_step&quot;</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span>
10561065

10571066
<span class="c1"># Evaluation loop</span>
10581067
<span class="k">with</span> <span class="n">torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">():</span>
@@ -1107,6 +1116,26 @@ <h1>Source code for perceptionmetrics.models.torch_segmentation</h1><div class="
11071116
<span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">predictions_outdir</span><span class="p">,</span> <span class="sa">f</span><span class="s2">&quot;</span><span class="si">{</span><span class="n">sample_name</span><span class="si">}</span><span class="s2">.bin&quot;</span><span class="p">)</span>
11081117
<span class="p">)</span>
11091118

1119+
<span class="n">processed_samples</span> <span class="o">+=</span> <span class="mi">1</span>
1120+
1121+
<span class="k">if</span> <span class="n">progress_callback</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
1122+
<span class="n">progress_callback</span><span class="p">(</span><span class="n">processed_samples</span><span class="p">,</span> <span class="n">total_samples</span><span class="p">)</span>
1123+
1124+
<span class="k">if</span> <span class="p">(</span>
1125+
<span class="n">metrics_callback</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span>
1126+
<span class="ow">and</span> <span class="n">evaluation_step</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span>
1127+
<span class="ow">and</span> <span class="n">evaluation_step</span> <span class="o">&gt;</span> <span class="mi">0</span>
1128+
<span class="ow">and</span> <span class="p">(</span>
1129+
<span class="n">processed_samples</span> <span class="o">%</span> <span class="n">evaluation_step</span> <span class="o">==</span> <span class="mi">0</span>
1130+
<span class="ow">or</span> <span class="n">processed_samples</span> <span class="o">==</span> <span class="n">total_samples</span>
1131+
<span class="p">)</span>
1132+
<span class="p">):</span>
1133+
<span class="n">metrics_callback</span><span class="p">(</span>
1134+
<span class="n">um</span><span class="o">.</span><span class="n">get_metrics_dataframe</span><span class="p">(</span><span class="n">metrics_factory</span><span class="p">,</span> <span class="n">eval_ontology</span><span class="p">),</span>
1135+
<span class="n">processed_samples</span><span class="p">,</span>
1136+
<span class="n">total_samples</span><span class="p">,</span>
1137+
<span class="p">)</span>
1138+
11101139
<span class="k">return</span> <span class="n">um</span><span class="o">.</span><span class="n">get_metrics_dataframe</span><span class="p">(</span><span class="n">metrics_factory</span><span class="p">,</span> <span class="n">eval_ontology</span><span class="p">)</span></div>
11111140

11121141

docs/py_docs/build/html/api/perceptionmetrics.models.html

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1421,7 +1421,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Link to this headi
14211421
</dl>
14221422
<dl class="py method">
14231423
<dt class="sig sig-object py" id="perceptionmetrics.models.torch_segmentation.TorchLiDARSegmentationModel.eval">
1424-
<span class="sig-name descname"><span class="pre">eval</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">dataset</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">split</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'test'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">ontology_translation</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">translation_direction</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'dataset_to_model'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">predictions_outdir</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">results_per_sample</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/perceptionmetrics/models/torch_segmentation.html#TorchLiDARSegmentationModel.eval"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#perceptionmetrics.models.torch_segmentation.TorchLiDARSegmentationModel.eval" title="Link to this definition"></a></dt>
1424+
<span class="sig-name descname"><span class="pre">eval</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">dataset</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">split</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'test'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">ontology_translation</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">translation_direction</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'dataset_to_model'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">predictions_outdir</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">results_per_sample</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">progress_callback</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">metrics_callback</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/perceptionmetrics/models/torch_segmentation.html#TorchLiDARSegmentationModel.eval"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#perceptionmetrics.models.torch_segmentation.TorchLiDARSegmentationModel.eval" title="Link to this definition"></a></dt>
14251425
<dd><p>Perform evaluation for a LiDAR segmentation dataset</p>
14261426
<dl class="field-list simple">
14271427
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -1432,6 +1432,8 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Link to this headi
14321432
<li><p><strong>translation_direction</strong> (<em>str</em><em>, </em><em>optional</em>) – Direction of the ontology translation, either ‘dataset_to_model’ or ‘model_to_dataset’, defaults to “dataset_to_model”</p></li>
14331433
<li><p><strong>predictions_outdir</strong> (<em>Optional</em><em>[</em><em>str</em><em>]</em><em>, </em><em>optional</em>) – Directory to save predictions per sample, defaults to None. If None, predictions are not saved.</p></li>
14341434
<li><p><strong>results_per_sample</strong> (<em>bool</em><em>, </em><em>optional</em>) – Whether to store results per sample or not, defaults to False. If True, predictions_outdir must be provided.</p></li>
1435+
<li><p><strong>progress_callback</strong> (<em>Optional</em><em>[</em><em>Callable</em><em>]</em><em>, </em><em>optional</em>) – Optional callback called as progress_callback(processed, total), defaults to None</p></li>
1436+
<li><p><strong>metrics_callback</strong> (<em>Optional</em><em>[</em><em>Callable</em><em>]</em><em>, </em><em>optional</em>) – Optional callback called as metrics_callback(metrics_df, processed, total), defaults to None</p></li>
14351437
</ul>
14361438
</dd>
14371439
<dt class="field-even">Returns<span class="colon">:</span></dt>

0 commit comments

Comments
 (0)