ImageBind Library

__init__.py

@@ -0,0 +1,3 @@
+from .data import load_and_transform_text, load_and_transform_vision_data, load_and_transform_audio_data
+from .models import imagebind_model
+from .models.imagebind_model import ModalityType

requirements.txt

@@ -1,4 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu113
 torch==1.13.0
 torchvision==0.14.0
 torchaudio==0.13.0
@@ -10,7 +9,7 @@ einops
 fvcore
 decord==0.6.0
 iopath
-numpy
+numpy>=1.19
 matplotlib
 types-regex
 mayavi

setup.py

@@ -0,0 +1,20 @@
+from setuptools import setup, find_packages
+
+with open('requirements.txt') as f:
+    required = f.read().splitlines()
+
+setup(
+    name='imagebind',
+    version='0.1.0',
+    packages=find_packages(),
+    description='A brief description of the package',
+    long_description=open('README.md').read(),
+    long_description_content_type="text/markdown",
+    url='https://github.com/facebookresearch/ImageBind',
+    classifiers=[
+        'Programming Language :: Python :: 3',
+        'License :: Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International',
+    ],
+    install_requires=required,
+    dependency_links=['https://download.pytorch.org/whl/cu113'],
+)