Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
TencentARC
GitHub Repository: TencentARC/GFPGAN
 Path: blob/master/setup.py
884 views
1
#!/usr/bin/env python

2


3
from setuptools import find_packages, setup

4


5
import os

6
import subprocess

7
import time

8


9
version_file = 'gfpgan/version.py'

10


11


12
def readme():

13
    with open('README.md', encoding='utf-8') as f:

14
        content = f.read()

15
    return content

16


17


18
def get_git_hash():

19


20
    def _minimal_ext_cmd(cmd):

21
        # construct minimal environment

22
        env = {}

23
        for k in ['SYSTEMROOT', 'PATH', 'HOME']:

24
            v = os.environ.get(k)

25
            if v is not None:

26
                env[k] = v

27
        # LANGUAGE is used on win32

28
        env['LANGUAGE'] = 'C'

29
        env['LANG'] = 'C'

30
        env['LC_ALL'] = 'C'

31
        out = subprocess.Popen(cmd, stdout=subprocess.PIPE, env=env).communicate()[0]

32
        return out

33


34
    try:

35
        out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])

36
        sha = out.strip().decode('ascii')

37
    except OSError:

38
        sha = 'unknown'

39


40
    return sha

41


42


43
def get_hash():

44
    if os.path.exists('.git'):

45
        sha = get_git_hash()[:7]

46
    else:

47
        sha = 'unknown'

48


49
    return sha

50


51


52
def write_version_py():

53
    content = """# GENERATED VERSION FILE

54
# TIME: {}

55
__version__ = '{}'

56
__gitsha__ = '{}'

57
version_info = ({})

58
"""

59
    sha = get_hash()

60
    with open('VERSION', 'r') as f:

61
        SHORT_VERSION = f.read().strip()

62
    VERSION_INFO = ', '.join([x if x.isdigit() else f'"{x}"' for x in SHORT_VERSION.split('.')])

63


64
    version_file_str = content.format(time.asctime(), SHORT_VERSION, sha, VERSION_INFO)

65
    with open(version_file, 'w') as f:

66
        f.write(version_file_str)

67


68


69
def get_version():

70
    with open(version_file, 'r') as f:

71
        exec(compile(f.read(), version_file, 'exec'))

72
    return locals()['__version__']

73


74


75
def get_requirements(filename='requirements.txt'):

76
    here = os.path.dirname(os.path.realpath(__file__))

77
    with open(os.path.join(here, filename), 'r') as f:

78
        requires = [line.replace('\n', '') for line in f.readlines()]

79
    return requires

80


81


82
if __name__ == '__main__':

83
    write_version_py()

84
    setup(

85
        name='gfpgan',

86
        version=get_version(),

87
        description='GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration',

88
        long_description=readme(),

89
        long_description_content_type='text/markdown',

90
        author='Xintao Wang',

91
        author_email='[email protected]',

92
        keywords='computer vision, pytorch, image restoration, super-resolution, face restoration, gan, gfpgan',

93
        url='https://github.com/TencentARC/GFPGAN',

94
        include_package_data=True,

95
        packages=find_packages(exclude=('options', 'datasets', 'experiments', 'results', 'tb_logger', 'wandb')),

96
        classifiers=[

97
            'Development Status :: 4 - Beta',

98
            'License :: OSI Approved :: Apache Software License',

99
            'Operating System :: OS Independent',

100
            'Programming Language :: Python :: 3',

101
            'Programming Language :: Python :: 3.7',

102
            'Programming Language :: Python :: 3.8',

103
        ],

104
        license='Apache License Version 2.0',

105
        setup_requires=['cython', 'numpy'],

106
        install_requires=get_requirements(),

107
        zip_safe=False)

108


109