-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
42 lines (37 loc) · 1.32 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import os
import sys
import shutil
from setuptools import setup
def clean_repo():
repo_folder = os.path.realpath(os.path.dirname(__file__))
dist_folder = os.path.join(repo_folder, 'dist')
build_folder = os.path.join(repo_folder, 'build')
if os.path.isdir(dist_folder):
shutil.rmtree(dist_folder, ignore_errors=True)
if os.path.isdir(build_folder):
shutil.rmtree(build_folder, ignore_errors=True)
# Read version string
_version = None
script_folder = os.path.realpath(os.path.dirname(__file__))
with open(os.path.join(script_folder, 'ibug', 'face_embedding', '__init__.py')) as init:
for line in init.read().splitlines():
fields = line.replace('=', ' ').replace('\'', ' ').replace('\"', ' ').replace('\t', ' ').split()
if len(fields) >= 2 and fields[0] == '__version__':
_version = fields[1]
break
if _version is None:
sys.exit('Sorry, cannot find version information.')
# Installation
config = {
'name': 'ibug_face_embedding',
'version': _version,
'description': 'Facial embedding models.',
'author': 'Jie Shen',
'author_email': 'js1907@imperial.ac.uk',
'packages': ['ibug.face_embedding'],
'install_requires': ['numpy>=1.16.0', 'torch>=1.1.0', 'opencv-python>=3.4.2'],
'zip_safe': False
}
clean_repo()
setup(**config)
clean_repo()