From e745f6ce0afa80e09f7fce3f4fde3e1b26a9eaf9 Mon Sep 17 00:00:00 2001 From: Mateen Ulhaq Date: Mon, 17 Jun 2024 22:12:53 -0700 Subject: [PATCH] refactor: rename compressai -> src --- {compressai => src}/__init__.py | 0 {compressai => src}/entropy_models/__init__.py | 0 {compressai => src}/entropy_models/entropy_models.py | 0 {compressai => src}/latent_codecs/__init__.py | 0 {compressai => src}/latent_codecs/entropy_bottleneck.py | 0 {compressai => src}/layers/__init__.py | 0 {compressai => src}/layers/basic.py | 0 {compressai => src}/layers/hist.py | 0 {compressai => src}/losses/__init__.py | 0 {compressai => src}/losses/rate_distortion.py | 0 {compressai => src}/models/__init__.py | 0 {compressai => src}/models/adaptive.py | 9 +++++---- {compressai => src}/ops/__init__.py | 0 {compressai => src}/ops/bound_ops.py | 0 {compressai => src}/ops/ops.py | 0 {compressai => src}/optimizers/__init__.py | 0 {compressai => src}/optimizers/net_aux.py | 0 {compressai => src}/utils/__init__.py | 0 {compressai => src}/utils/plot/__init__.py | 0 {compressai => src}/utils/plot/__main__.py | 0 20 files changed, 5 insertions(+), 4 deletions(-) rename {compressai => src}/__init__.py (100%) rename {compressai => src}/entropy_models/__init__.py (100%) rename {compressai => src}/entropy_models/entropy_models.py (100%) rename {compressai => src}/latent_codecs/__init__.py (100%) rename {compressai => src}/latent_codecs/entropy_bottleneck.py (100%) rename {compressai => src}/layers/__init__.py (100%) rename {compressai => src}/layers/basic.py (100%) rename {compressai => src}/layers/hist.py (100%) rename {compressai => src}/losses/__init__.py (100%) rename {compressai => src}/losses/rate_distortion.py (100%) rename {compressai => src}/models/__init__.py (100%) rename {compressai => src}/models/adaptive.py (99%) rename {compressai => src}/ops/__init__.py (100%) rename {compressai => src}/ops/bound_ops.py (100%) rename {compressai => src}/ops/ops.py (100%) rename {compressai => src}/optimizers/__init__.py (100%) rename {compressai => src}/optimizers/net_aux.py (100%) rename {compressai => src}/utils/__init__.py (100%) rename {compressai => src}/utils/plot/__init__.py (100%) rename {compressai => src}/utils/plot/__main__.py (100%) diff --git a/compressai/__init__.py b/src/__init__.py similarity index 100% rename from compressai/__init__.py rename to src/__init__.py diff --git a/compressai/entropy_models/__init__.py b/src/entropy_models/__init__.py similarity index 100% rename from compressai/entropy_models/__init__.py rename to src/entropy_models/__init__.py diff --git a/compressai/entropy_models/entropy_models.py b/src/entropy_models/entropy_models.py similarity index 100% rename from compressai/entropy_models/entropy_models.py rename to src/entropy_models/entropy_models.py diff --git a/compressai/latent_codecs/__init__.py b/src/latent_codecs/__init__.py similarity index 100% rename from compressai/latent_codecs/__init__.py rename to src/latent_codecs/__init__.py diff --git a/compressai/latent_codecs/entropy_bottleneck.py b/src/latent_codecs/entropy_bottleneck.py similarity index 100% rename from compressai/latent_codecs/entropy_bottleneck.py rename to src/latent_codecs/entropy_bottleneck.py diff --git a/compressai/layers/__init__.py b/src/layers/__init__.py similarity index 100% rename from compressai/layers/__init__.py rename to src/layers/__init__.py diff --git a/compressai/layers/basic.py b/src/layers/basic.py similarity index 100% rename from compressai/layers/basic.py rename to src/layers/basic.py diff --git a/compressai/layers/hist.py b/src/layers/hist.py similarity index 100% rename from compressai/layers/hist.py rename to src/layers/hist.py diff --git a/compressai/losses/__init__.py b/src/losses/__init__.py similarity index 100% rename from compressai/losses/__init__.py rename to src/losses/__init__.py diff --git a/compressai/losses/rate_distortion.py b/src/losses/rate_distortion.py similarity index 100% rename from compressai/losses/rate_distortion.py rename to src/losses/rate_distortion.py diff --git a/compressai/models/__init__.py b/src/models/__init__.py similarity index 100% rename from compressai/models/__init__.py rename to src/models/__init__.py diff --git a/compressai/models/adaptive.py b/src/models/adaptive.py similarity index 99% rename from compressai/models/adaptive.py rename to src/models/adaptive.py index 8086f55..a68ed28 100644 --- a/compressai/models/adaptive.py +++ b/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 b/src/ops/__init__.py similarity index 100% rename from compressai/ops/__init__.py rename to src/ops/__init__.py diff --git a/compressai/ops/bound_ops.py b/src/ops/bound_ops.py similarity index 100% rename from compressai/ops/bound_ops.py rename to src/ops/bound_ops.py diff --git a/compressai/ops/ops.py b/src/ops/ops.py similarity index 100% rename from compressai/ops/ops.py rename to src/ops/ops.py diff --git a/compressai/optimizers/__init__.py b/src/optimizers/__init__.py similarity index 100% rename from compressai/optimizers/__init__.py rename to src/optimizers/__init__.py diff --git a/compressai/optimizers/net_aux.py b/src/optimizers/net_aux.py similarity index 100% rename from compressai/optimizers/net_aux.py rename to src/optimizers/net_aux.py diff --git a/compressai/utils/__init__.py b/src/utils/__init__.py similarity index 100% rename from compressai/utils/__init__.py rename to src/utils/__init__.py diff --git a/compressai/utils/plot/__init__.py b/src/utils/plot/__init__.py similarity index 100% rename from compressai/utils/plot/__init__.py rename to src/utils/plot/__init__.py diff --git a/compressai/utils/plot/__main__.py b/src/utils/plot/__main__.py similarity index 100% rename from compressai/utils/plot/__main__.py rename to src/utils/plot/__main__.py