Skip to content

Commit

Permalink
deploy: 8921825
Browse files Browse the repository at this point in the history
  • Loading branch information
@fracape
fracape committed Feb 28, 2024
1 parent 53a07b5 commit ed3a6f1
Show file tree
Hide file tree
Showing 119 changed files with 13,022 additions and 1,782 deletions.
1 change: 1 addition & 0 deletions _modules/compressai.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/datasets/image.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
540 changes: 540 additions & 0 deletions _modules/compressai/datasets/pointcloud/modelnet.html
View file

Large diffs are not rendered by default.

562 changes: 562 additions & 0 deletions _modules/compressai/datasets/pointcloud/s3dis.html
View file

Large diffs are not rendered by default.

626 changes: 626 additions & 0 deletions _modules/compressai/datasets/pointcloud/semantic_kitti.html
View file

Large diffs are not rendered by default.

636 changes: 636 additions & 0 deletions _modules/compressai/datasets/pointcloud/shapenet.html
View file

Large diffs are not rendered by default.

465 changes: 465 additions & 0 deletions _modules/compressai/datasets/pregenerated.html
View file

Large diffs are not rendered by default.

1 change: 1 addition & 0 deletions _modules/compressai/datasets/video.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
461 changes: 461 additions & 0 deletions _modules/compressai/datasets/vimeo90k.html
View file

Large diffs are not rendered by default.

25 changes: 22 additions & 3 deletions _modules/compressai/entropy_models/entropy_models.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down Expand Up @@ -675,7 +676,7 @@ <h1>Source code for compressai.entropy_models.entropy_models</h1><div class="hig
<span class="n">medians</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">quantiles</span><span class="p">[:,</span> <span class="p">:,</span> <span class="mi">1</span><span class="p">:</span><span class="mi">2</span><span class="p">]</span>
<span class="k">return</span> <span class="n">medians</span>

<span class="k">def</span> <span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">force</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> <span class="n">update_quantiles</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
<span class="k">def</span> <span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">force</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> <span class="n">update_quantiles</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">bool</span><span class="p">:</span>
<span class="c1"># Check if we need to update the bottleneck parameters, the offsets are</span>
<span class="c1"># only computed and stored when the conditonal model is update()&#39;d.</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_offset</span><span class="o">.</span><span class="n">numel</span><span class="p">()</span> <span class="o">&gt;</span> <span class="mi">0</span> <span class="ow">and</span> <span class="ow">not</span> <span class="n">force</span><span class="p">:</span>
Expand Down Expand Up @@ -815,6 +816,10 @@ <h1>Source code for compressai.entropy_models.entropy_models</h1><div class="hig

<span class="nd">@torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">()</span>
<span class="k">def</span> <span class="nf">_update_quantiles</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">search_radius</span><span class="o">=</span><span class="mf">1e5</span><span class="p">,</span> <span class="n">rtol</span><span class="o">=</span><span class="mf">1e-4</span><span class="p">,</span> <span class="n">atol</span><span class="o">=</span><span class="mf">1e-3</span><span class="p">):</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Fast quantile update via bisection search.</span>

<span class="sd"> Often faster and much more precise than minimizing aux loss.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">device</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">quantiles</span><span class="o">.</span><span class="n">device</span>
<span class="n">shape</span> <span class="o">=</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">channels</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span>
<span class="n">low</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">full</span><span class="p">(</span><span class="n">shape</span><span class="p">,</span> <span class="o">-</span><span class="n">search_radius</span><span class="p">,</span> <span class="n">device</span><span class="o">=</span><span class="n">device</span><span class="p">)</span>
Expand Down Expand Up @@ -1173,10 +1178,17 @@ <h1>Source code for compressai.entropy_models.entropy_models</h1><div class="hig

<span class="n">num_latents</span> <span class="o">=</span> <span class="n">cdf</span><span class="o">.</span><span class="n">size</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>

<span class="c1"># rv = self.entropy_coder._encoder.encode_with_indexes(</span>
<span class="c1"># symbols.reshape(-1).int().tolist(),</span>
<span class="c1"># torch.arange(num_latents).int().tolist(),</span>
<span class="c1"># cdf.cpu().to(torch.int32),</span>
<span class="c1"># torch.tensor(cdf.size(1)).repeat(num_latents).int().tolist(),</span>
<span class="c1"># torch.tensor(0).repeat(num_latents).int().tolist(),</span>
<span class="c1"># )</span>
<span class="n">rv</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">entropy_coder</span><span class="o">.</span><span class="n">_encoder</span><span class="o">.</span><span class="n">encode_with_indexes</span><span class="p">(</span>
<span class="n">symbols</span><span class="o">.</span><span class="n">reshape</span><span class="p">(</span><span class="o">-</span><span class="mi">1</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">torch</span><span class="o">.</span><span class="n">arange</span><span class="p">(</span><span class="n">num_latents</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">cdf</span><span class="o">.</span><span class="n">cpu</span><span class="p">()</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">int32</span><span class="p">),</span>
<span class="n">cdf</span><span class="o">.</span><span class="n">cpu</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">(</span><span class="n">cdf</span><span class="o">.</span><span class="n">size</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span><span class="o">.</span><span class="n">repeat</span><span class="p">(</span><span class="n">num_latents</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span><span class="o">.</span><span class="n">repeat</span><span class="p">(</span><span class="n">num_latents</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="p">)</span>
Expand All @@ -1191,10 +1203,17 @@ <h1>Source code for compressai.entropy_models.entropy_models</h1><div class="hig

<span class="n">num_latents</span> <span class="o">=</span> <span class="n">cdf</span><span class="o">.</span><span class="n">size</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>

<span class="c1"># values = self.entropy_coder._decoder.decode_with_indexes(</span>
<span class="c1"># strings,</span>
<span class="c1"># torch.arange(num_latents).int().tolist(),</span>
<span class="c1"># cdf.cpu().to(torch.int32),</span>
<span class="c1"># torch.tensor(cdf.size(1)).repeat(num_latents).int().tolist(),</span>
<span class="c1"># torch.tensor(0).repeat(num_latents).int().tolist(),</span>
<span class="c1"># )</span>
<span class="n">values</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">entropy_coder</span><span class="o">.</span><span class="n">_decoder</span><span class="o">.</span><span class="n">decode_with_indexes</span><span class="p">(</span>
<span class="n">strings</span><span class="p">,</span>
<span class="n">torch</span><span class="o">.</span><span class="n">arange</span><span class="p">(</span><span class="n">num_latents</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">cdf</span><span class="o">.</span><span class="n">cpu</span><span class="p">()</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">int32</span><span class="p">),</span>
<span class="n">cdf</span><span class="o">.</span><span class="n">cpu</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">(</span><span class="n">cdf</span><span class="o">.</span><span class="n">size</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span><span class="o">.</span><span class="n">repeat</span><span class="p">(</span><span class="n">num_latents</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="n">torch</span><span class="o">.</span><span class="n">tensor</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span><span class="o">.</span><span class="n">repeat</span><span class="p">(</span><span class="n">num_latents</span><span class="p">)</span><span class="o">.</span><span class="n">int</span><span class="p">()</span><span class="o">.</span><span class="n">tolist</span><span class="p">(),</span>
<span class="p">)</span>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/channel_groups.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/checkerboard.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/entropy_bottleneck.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/gain/hyper.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/gain/hyperprior.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/gaussian_conditional.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/hyper.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
1 change: 1 addition & 0 deletions _modules/compressai/latent_codecs/hyperprior.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,7 @@
<li class="toctree-l1"><a class="reference internal" href="../../../entropy_models.html">compressai.entropy_models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../latent_codecs.html">compressai.latent_codecs</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../layers.html">compressai.layers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../losses.html">compressai.losses</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../models.html">compressai.models</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../ops.html">compressai.ops</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../transforms.html">compressai.transforms</a></li>
Expand Down
Loading

0 comments on commit ed3a6f1

Please sign in to comment.