refactor: rename compressai -> src

multimedialabsfu · Jun 18, 2024 · e745f6c · e745f6c
1 parent d83fa2b
commit e745f6c
Show file tree

Hide file tree

Showing 20 changed files with 5 additions and 4 deletions.
diff --git a/compressai/__init__.py → src/__init__.py b/compressai/__init__.py → src/__init__.py
diff --git a/compressai/entropy_models/__init__.py → src/entropy_models/__init__.py b/compressai/entropy_models/__init__.py → src/entropy_models/__init__.py
diff --git a/compressai/entropy_models/entropy_models.py → src/entropy_models/entropy_models.py b/compressai/entropy_models/entropy_models.py → src/entropy_models/entropy_models.py
diff --git a/compressai/latent_codecs/__init__.py → src/latent_codecs/__init__.py b/compressai/latent_codecs/__init__.py → src/latent_codecs/__init__.py
diff --git a/...essai/latent_codecs/entropy_bottleneck.py → src/latent_codecs/entropy_bottleneck.py b/...essai/latent_codecs/entropy_bottleneck.py → src/latent_codecs/entropy_bottleneck.py
diff --git a/compressai/layers/__init__.py → src/layers/__init__.py b/compressai/layers/__init__.py → src/layers/__init__.py
diff --git a/compressai/layers/basic.py → src/layers/basic.py b/compressai/layers/basic.py → src/layers/basic.py
diff --git a/compressai/layers/hist.py → src/layers/hist.py b/compressai/layers/hist.py → src/layers/hist.py
diff --git a/compressai/losses/__init__.py → src/losses/__init__.py b/compressai/losses/__init__.py → src/losses/__init__.py
diff --git a/compressai/losses/rate_distortion.py → src/losses/rate_distortion.py b/compressai/losses/rate_distortion.py → src/losses/rate_distortion.py
diff --git a/compressai/models/__init__.py → src/models/__init__.py b/compressai/models/__init__.py → src/models/__init__.py
diff --git a/compressai/models/adaptive.py → src/models/adaptive.py b/compressai/models/adaptive.py → src/models/adaptive.py
@@ -2,20 +2,21 @@
 
 import torch
 import torch.nn as nn
-
-from compressai.entropy_models.entropy_models import EntropyBottleneck, pdf_layout
+from compressai.entropy_models.entropy_models import EntropyBottleneck
 from compressai.latent_codecs import LatentCodec
-from compressai.layers import UniformHistogram
 from compressai.models.base import CompressionModel
 from compressai.models.google import (
     FactorizedPrior,
     MeanScaleHyperprior,
     ScaleHyperprior,
 )
-from compressai.ops import RangeBound
 from compressai.registry import register_model, register_module
 from compressai.registry.torch import MODELS
 
+from src.entropy_models.entropy_models import pdf_layout
+from src.layers import UniformHistogram
+from src.ops import RangeBound
+
 
 class FreezeMixin:
     def _freeze_modules(self, freeze_modules, unfreeze_modules):

diff --git a/compressai/ops/__init__.py → src/ops/__init__.py b/compressai/ops/__init__.py → src/ops/__init__.py
diff --git a/compressai/ops/bound_ops.py → src/ops/bound_ops.py b/compressai/ops/bound_ops.py → src/ops/bound_ops.py
diff --git a/compressai/ops/ops.py → src/ops/ops.py b/compressai/ops/ops.py → src/ops/ops.py
diff --git a/compressai/optimizers/__init__.py → src/optimizers/__init__.py b/compressai/optimizers/__init__.py → src/optimizers/__init__.py
diff --git a/compressai/optimizers/net_aux.py → src/optimizers/net_aux.py b/compressai/optimizers/net_aux.py → src/optimizers/net_aux.py
diff --git a/compressai/utils/__init__.py → src/utils/__init__.py b/compressai/utils/__init__.py → src/utils/__init__.py
diff --git a/compressai/utils/plot/__init__.py → src/utils/plot/__init__.py b/compressai/utils/plot/__init__.py → src/utils/plot/__init__.py
diff --git a/compressai/utils/plot/__main__.py → src/utils/plot/__main__.py b/compressai/utils/plot/__main__.py → src/utils/plot/__main__.py