deploy: 99c03b7

_modules/tracklab/datastruct/tracker_state.html

@@ -207,9 +207,12 @@ <h1>Source code for tracklab.datastruct.tracker_state</h1><div class="highlight"
                 <span class="n">load_columns</span> <span class="o">=</span> <span class="p">{</span><span class="n">k</span><span class="p">:</span> <span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">load_from_groundtruth</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">k</span><span class="p">,</span> <span class="p">[]))</span> <span class="o">&amp;</span> <span class="nb">set</span><span class="p">(</span><span class="n">v</span><span class="p">))</span>
                                 <span class="k">for</span> <span class="n">k</span><span class="p">,</span> <span class="n">v</span> <span class="ow">in</span> <span class="n">load_columns</span><span class="o">.</span><span class="n">items</span><span class="p">()</span>
                                 <span class="p">}</span>
-            <span class="bp">self</span><span class="o">.</span><span class="n">detections_pred_gt</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">detections_gt</span><span class="o">.</span><span class="n">copy</span><span class="p">()[</span>
-                <span class="bp">self</span><span class="o">.</span><span class="n">detections_gt</span><span class="o">.</span><span class="n">columns</span><span class="o">.</span><span class="n">intersection</span><span class="p">(</span><span class="n">load_columns</span><span class="p">[</span><span class="s2">&quot;detection&quot;</span><span class="p">]</span><span class="o">+</span><span class="p">[</span><span class="s2">&quot;image_id&quot;</span><span class="p">,</span> <span class="s2">&quot;video_id&quot;</span><span class="p">])</span>
-            <span class="p">]</span>
+            <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">load_columns</span><span class="p">[</span><span class="s2">&quot;detection&quot;</span><span class="p">])</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
+                <span class="bp">self</span><span class="o">.</span><span class="n">detections_pred_gt</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">columns</span><span class="o">=</span><span class="p">[</span><span class="s2">&quot;video_id&quot;</span><span class="p">,</span><span class="s2">&quot;image_id&quot;</span><span class="p">])</span>
+            <span class="k">else</span><span class="p">:</span>
+                <span class="bp">self</span><span class="o">.</span><span class="n">detections_pred_gt</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">detections_gt</span><span class="o">.</span><span class="n">copy</span><span class="p">()[</span>
+                    <span class="bp">self</span><span class="o">.</span><span class="n">detections_gt</span><span class="o">.</span><span class="n">columns</span><span class="o">.</span><span class="n">intersection</span><span class="p">(</span><span class="n">load_columns</span><span class="p">[</span><span class="s2">&quot;detection&quot;</span><span class="p">]</span><span class="o">+</span><span class="p">[</span><span class="s2">&quot;image_id&quot;</span><span class="p">,</span> <span class="s2">&quot;video_id&quot;</span><span class="p">])</span>
+                <span class="p">]</span>
             <span class="bp">self</span><span class="o">.</span><span class="n">image_pred_gt</span> <span class="o">=</span> <span class="n">merge_dataframes</span><span class="p">(</span>
                 <span class="bp">self</span><span class="o">.</span><span class="n">image_metadatas</span><span class="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="bp">self</span><span class="o">.</span><span class="n">image_gt</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span>
             <span class="p">)[</span><span class="nb">list</span><span class="p">(</span><span class="nb">set</span><span class="p">(</span><span class="n">load_columns</span><span class="p">[</span><span class="s2">&quot;image&quot;</span><span class="p">])</span> <span class="o">|</span> <span class="p">{</span><span class="s2">&quot;video_id&quot;</span><span class="p">,</span> <span class="s2">&quot;file_path&quot;</span><span class="p">,</span> <span class="s2">&quot;frame&quot;</span><span class="p">})]</span></div>