models and structure changes

adriandavidauer · Mar 31, 2021 · cd72358 · cd72358
1 parent e6a537f
commit cd72358
Show file tree

Hide file tree

Showing 12 changed files with 660 additions and 472 deletions.
diff --git a/setup.py b/setup.py
@@ -10,9 +10,7 @@
 # local imports
 
 # end file header
-__author__      = 'Adrian Lubitz'
-
-
+__author__ = 'Adrian Lubitz'
 
 
 # Utility function to read the README file.
@@ -22,20 +20,22 @@
 def read(fname):
     return open(os.path.join(os.path.dirname(__file__), fname)).read()
 
+
 if 'RELEASE_VERSION' in os.environ:
     VERSION = os.environ['RELEASE_VERSION']
 else:
     VERSION = '0.0.0'
 
 setup(
-    name = "vvadlrs3",
-    version = VERSION, # TODO: version should be set from tag or whatever
-    author = "Adrian Lubitz",
-    author_email = "adrianlubitz@gmail.com",
-    description = ("Library to provide models trained on the VVAD-LRS3 Dataset. The library also contains preprocessing pipelines."),
-    license = "LGPLv2. Note that the license for the iBUG 300-W dataset which was used for face and lip features excludes commercial use.",
-    keywords = "VVAD LRS3 AI Social robotics",
-    url = "https://github.com/adrianlubitz/VVAD",
+    name="vvadlrs3",
+    version=VERSION,  # TODO: version should be set from tag or whatever
+    author="Adrian Lubitz",
+    author_email="adrianlubitz@gmail.com",
+    description=(
+        "Library to provide models trained on the VVAD-LRS3 Dataset. The library also contains preprocessing pipelines."),
+    license="LGPLv2. Note that the license for the iBUG 300-W dataset which was used for face and lip features excludes commercial use.",
+    keywords="VVAD LRS3 AI Social robotics",
+    url="https://github.com/adrianlubitz/VVAD",
     packages=['vvadlrs3'],
     package_data={
         # Include all h5 files in the vvadlrs3 package
@@ -49,12 +49,13 @@ def read(fname):
         "License :: OSI Approved :: GNU Lesser General Public License v2 (LGPLv2)",
     ],
     install_requires=[
-          'numpy', # TODO: Which are the needed dependencies
-          'opencv-contrib-python',
-          'dlib', # TODO: This guy is problematic because it needs cmake...That needs to be communicated somewhere
-          'matplotlib',
-          'keras==2.4.3', 
-          'tensorflow==2.3.1'
-      ],
+        'numpy',  # TODO: Which are the needed dependencies
+        'opencv-contrib-python',
+        'dlib',  # TODO: This guy is problematic because it needs cmake...That needs to be communicated somewhere
+        'matplotlib',
+        'keras==2.4.3',
+        'tensorflow==2.3.1',
+        'tqdm'
+    ],
     python_requires='>=3.7',
-)
+)