google-research · jayathungek · Sep 29, 2022 · Sep 29, 2022 · Sep 29, 2022 · Sep 29, 2022
diff --git a/setup.py b/setup.py
@@ -0,0 +1,28 @@
+from setuptools import setup, find_packages
+
+from version import version
+
+
+setup_info = dict(
+    name='simclr',
+    version=version,
+    author='google-research',
+    url='https://github.com/google-research/simclr',
+    download_url='https://github.com/google-research/simclr/tags',
+    project_urls={
+        'Source': 'https://github.com/google-research/simclr',
+    },
+    description='A Simple Framework for Contrastive Learning of Visual Representations',
+    long_description=open('README.md').read(),
+    long_description_content_type='text/markdown',
+    license='Apache License Version 2.0',
+
+    # Package info
+    packages=['simclr'] + ['simclr.' + pkg for pkg in find_packages('simclr')],
+
+    # Add _ prefix to the names of temporary build dirs
+    options={'build': {'build_base': '_build'}, },
+    zip_safe=True,
+)
+
+setup(**setup_info)
diff --git a/simclr/__init__.py b/simclr/__init__.py
diff --git a/data.py → simclr/data.py b/data.py → simclr/data.py
@@ -22,7 +22,7 @@
 import functools
 from absl import flags
 
-import data_util as data_util
+import simclr.data_util as data_util
 import tensorflow.compat.v1 as tf
 
 FLAGS = flags.FLAGS

diff --git a/data_util.py → simclr/data_util.py b/data_util.py → simclr/data_util.py
diff --git a/lars_optimizer.py → simclr/lars_optimizer.py b/lars_optimizer.py → simclr/lars_optimizer.py
diff --git a/model.py → simclr/model.py b/model.py → simclr/model.py
@@ -21,9 +21,9 @@
 
 from absl import flags
 
-import data_util as data_util
-import model_util as model_util
-import objective as obj_lib
+import simclr.data_util as data_util
+import simclr.model_util as model_util
+import simclr.objective as obj_lib
 
 import tensorflow.compat.v1 as tf
 import tensorflow.compat.v2 as tf2

diff --git a/model_util.py → simclr/model_util.py b/model_util.py → simclr/model_util.py
@@ -22,7 +22,7 @@
 import math
 from absl import flags
 
-import resnet
+import simclr.resnet
 from lars_optimizer import LARSOptimizer
 
 import tensorflow.compat.v1 as tf

diff --git a/objective.py → simclr/objective.py b/objective.py → simclr/objective.py
diff --git a/resnet.py → simclr/resnet.py b/resnet.py → simclr/resnet.py
diff --git a/run.py → simclr/run.py b/run.py → simclr/run.py
@@ -25,10 +25,10 @@
 from absl import app
 from absl import flags
 
-import resnet
-import data as data_lib
-import model as model_lib
-import model_util as model_util
+import simclr.resnet as resnet
+import simclr.data as data_lib
+import simclr.model as model_lib
+import simclr.model_util as model_util
 
 import tensorflow.compat.v1 as tf
 import tensorflow_datasets as tfds

diff --git a/tf2/README.md → simclr/tf2/README.md b/tf2/README.md → simclr/tf2/README.md
diff --git a/simclr/tf2/__init__.py b/simclr/tf2/__init__.py
diff --git a/tf2/colabs/distillation_self_training.ipynb → ...2/colabs/distillation_self_training.ipynb b/tf2/colabs/distillation_self_training.ipynb → ...2/colabs/distillation_self_training.ipynb
diff --git a/tf2/colabs/finetuning.ipynb → simclr/tf2/colabs/finetuning.ipynb b/tf2/colabs/finetuning.ipynb → simclr/tf2/colabs/finetuning.ipynb
diff --git a/tf2/colabs/imagenet_results.ipynb → simclr/tf2/colabs/imagenet_results.ipynb b/tf2/colabs/imagenet_results.ipynb → simclr/tf2/colabs/imagenet_results.ipynb
diff --git a/tf2/colabs/load_and_inference.ipynb → simclr/tf2/colabs/load_and_inference.ipynb b/tf2/colabs/load_and_inference.ipynb → simclr/tf2/colabs/load_and_inference.ipynb
diff --git a/tf2/data.py → simclr/tf2/data.py b/tf2/data.py → simclr/tf2/data.py
@@ -19,7 +19,7 @@
 from absl import flags
 from absl import logging
 
-import data_util
+import simclr.tf2.data_util as data_util
 import tensorflow.compat.v2 as tf
 import tensorflow_datasets as tfds
 

diff --git a/tf2/data_util.py → simclr/tf2/data_util.py b/tf2/data_util.py → simclr/tf2/data_util.py
diff --git a/tf2/lars_optimizer.py → simclr/tf2/lars_optimizer.py b/tf2/lars_optimizer.py → simclr/tf2/lars_optimizer.py
diff --git a/tf2/metrics.py → simclr/tf2/metrics.py b/tf2/metrics.py → simclr/tf2/metrics.py
diff --git a/tf2/model.py → simclr/tf2/model.py b/tf2/model.py → simclr/tf2/model.py
@@ -18,9 +18,9 @@
 import math
 from absl import flags
 
-import data_util
-import lars_optimizer
-import resnet
+import simclr.tf2.data_util as data_util
+import simclr.tf2.lars_optimizer as lars_optimizer
+import simclr.tf2.resnet as resnet
 import tensorflow.compat.v2 as tf
 
 FLAGS = flags.FLAGS

diff --git a/tf2/objective.py → simclr/tf2/objective.py b/tf2/objective.py → simclr/tf2/objective.py
diff --git a/tf2/requirements.txt → simclr/tf2/requirements.txt b/tf2/requirements.txt → simclr/tf2/requirements.txt
diff --git a/tf2/resnet.py → simclr/tf2/resnet.py b/tf2/resnet.py → simclr/tf2/resnet.py
diff --git a/tf2/run.py → simclr/tf2/run.py b/tf2/run.py → simclr/tf2/run.py
@@ -22,10 +22,12 @@
 from absl import app
 from absl import flags
 from absl import logging
-import data as data_lib
-import metrics
-import model as model_lib
-import objective as obj_lib
+
+import simclr.tf2.data as data_lib
+import simclr.tf2.metrics as metrics
+import simclr.tf2.model as model_lib
+import simclr.tf2.objective as obj_lib
+
 import tensorflow.compat.v2 as tf
 import tensorflow_datasets as tfds
 

diff --git a/version.py b/version.py
@@ -0,0 +1 @@
+version = "1.0"