diff --git a/.github/workflows/pylint.yml b/.github/workflows/pylint.yml
index 1d8b501..d754f53 100644
--- a/.github/workflows/pylint.yml
+++ b/.github/workflows/pylint.yml
@@ -26,5 +26,5 @@ jobs:
     - name: Lint
       run: |
         flake8 .
-        isort --check-only --diff data/ models/ inference_realesrgan.py
-        yapf -r -d data/ models/ inference_realesrgan.py
+        isort --check-only --diff realesrgan/ scripts/ inference_realesrgan.py setup.py
+        yapf -r -d realesrgan/ scripts/ inference_realesrgan.py setup.py
diff --git a/archs/__init__.py b/realesrgan/archs/__init__.py
similarity index 100%
rename from archs/__init__.py
rename to realesrgan/archs/__init__.py
diff --git a/archs/discriminator_arch.py b/realesrgan/archs/discriminator_arch.py
similarity index 100%
rename from archs/discriminator_arch.py
rename to realesrgan/archs/discriminator_arch.py
diff --git a/data/__init__.py b/realesrgan/data/__init__.py
similarity index 100%
rename from data/__init__.py
rename to realesrgan/data/__init__.py
diff --git a/data/realesrgan_dataset.py b/realesrgan/data/realesrgan_dataset.py
similarity index 100%
rename from data/realesrgan_dataset.py
rename to realesrgan/data/realesrgan_dataset.py
diff --git a/models/__init__.py b/realesrgan/models/__init__.py
similarity index 100%
rename from models/__init__.py
rename to realesrgan/models/__init__.py
diff --git a/models/realesrgan_model.py b/realesrgan/models/realesrgan_model.py
similarity index 100%
rename from models/realesrgan_model.py
rename to realesrgan/models/realesrgan_model.py
diff --git a/models/realesrnet_model.py b/realesrgan/models/realesrnet_model.py
similarity index 100%
rename from models/realesrnet_model.py
rename to realesrgan/models/realesrnet_model.py
diff --git a/train.py b/realesrgan/train.py
similarity index 61%
rename from train.py
rename to realesrgan/train.py
index bd52322..d985a79 100644
--- a/train.py
+++ b/realesrgan/train.py
@@ -1,9 +1,9 @@
 import os.path as osp
 from basicsr.train import train_pipeline
 
-import archs  # noqa: F401
-import data  # noqa: F401
-import models  # noqa: F401
+import realesrgan.archs  # noqa: F401
+import realesrgan.data  # noqa: F401
+import realesrgan.models  # noqa: F401
 
 if __name__ == '__main__':
     root_path = osp.abspath(osp.join(__file__, osp.pardir))