dct-autoencoder 0.2.0__tar.gz → 0.2.1__tar.gz

{dct_autoencoder-0.2.0 → dct_autoencoder-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dct-autoencoder
-Version: 0.2.0
+Version: 0.2.1
 Summary: 
 Author: Dariush Bahrami
 Author-email: dariushbahrami1993@gmail.com

{dct_autoencoder-0.2.0 → dct_autoencoder-0.2.1}/dct_autoencoder/core.py

@@ -14,6 +14,10 @@ class DCTAutoencoder(nn.Module):
 
     Args:
         block_size (int, optional): The block size. Defaults to 8.
+        luminance_compression_ratio (float, optional): The luminance compression ratio.
+            Defaults to 1/2.
+        chrominance_compression_ratio (float, optional): The chrominance compression ratio.
+            Defaults to 1/4.
     """
 
     def __init__(
@@ -198,7 +202,7 @@ class DCTAutoencoder(nn.Module):
                 + 2 * self.compression_chrominance_passband.item()
             )
 
-    def compress(self, encodings):
+    def compress(self, encodings: torch.Tensor) -> torch.Tensor:
         if not self.do_compression:
             return encodings
         else:
@@ -211,7 +215,7 @@ class DCTAutoencoder(nn.Module):
             compressed_encoding = torch.cat([l, c1, c2], dim=1)
             return compressed_encoding
 
-    def decompress(self, compressed_encoding):
+    def decompress(self, compressed_encoding: torch.Tensor) -> torch.Tensor:
         if not self.do_compression:
             return compressed_encoding
         else:

{dct_autoencoder-0.2.0 → dct_autoencoder-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dct-autoencoder"
-version = "0.2.0"
+version = "0.2.1"
 description = ""
 authors = ["Dariush Bahrami <dariushbahrami1993@gmail.com>"]
 readme = "README.md"