| | |
| |
|
| | from setuptools import find_packages, setup |
| |
|
| | import os |
| | import subprocess |
| | import time |
| |
|
| | version_file = 'realesrgan/version.py' |
| |
|
| |
|
| | def readme(): |
| | with open('README.md', encoding='utf-8') as f: |
| | content = f.read() |
| | return content |
| |
|
| |
|
| | def get_git_hash(): |
| |
|
| | def _minimal_ext_cmd(cmd): |
| | |
| | env = {} |
| | for k in ['SYSTEMROOT', 'PATH', 'HOME']: |
| | v = os.environ.get(k) |
| | if v is not None: |
| | env[k] = v |
| | |
| | env['LANGUAGE'] = 'C' |
| | env['LANG'] = 'C' |
| | env['LC_ALL'] = 'C' |
| | out = subprocess.Popen(cmd, stdout=subprocess.PIPE, env=env).communicate()[0] |
| | return out |
| |
|
| | try: |
| | out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD']) |
| | sha = out.strip().decode('ascii') |
| | except OSError: |
| | sha = 'unknown' |
| |
|
| | return sha |
| |
|
| |
|
| | def get_hash(): |
| | if os.path.exists('.git'): |
| | sha = get_git_hash()[:7] |
| | else: |
| | sha = 'unknown' |
| |
|
| | return sha |
| |
|
| |
|
| | def write_version_py(): |
| | content = """# GENERATED VERSION FILE |
| | # TIME: {} |
| | __version__ = '{}' |
| | __gitsha__ = '{}' |
| | version_info = ({}) |
| | """ |
| | sha = get_hash() |
| | with open('VERSION', 'r') as f: |
| | SHORT_VERSION = f.read().strip() |
| | VERSION_INFO = ', '.join([x if x.isdigit() else f'"{x}"' for x in SHORT_VERSION.split('.')]) |
| |
|
| | version_file_str = content.format(time.asctime(), SHORT_VERSION, sha, VERSION_INFO) |
| | with open(version_file, 'w') as f: |
| | f.write(version_file_str) |
| |
|
| |
|
| | def get_version(): |
| | with open(version_file, 'r') as f: |
| | exec(compile(f.read(), version_file, 'exec')) |
| | return locals()['__version__'] |
| |
|
| |
|
| | def get_requirements(filename='requirements.txt'): |
| | here = os.path.dirname(os.path.realpath(__file__)) |
| | with open(os.path.join(here, filename), 'r') as f: |
| | requires = [line.replace('\n', '') for line in f.readlines()] |
| | return requires |
| |
|
| |
|
| | if __name__ == '__main__': |
| | write_version_py() |
| | setup( |
| | name='realesrgan', |
| | version=get_version(), |
| | description='Real-ESRGAN aims at developing Practical Algorithms for General Image Restoration', |
| | long_description=readme(), |
| | long_description_content_type='text/markdown', |
| | author='Xintao Wang', |
| | author_email='xintao.wang@outlook.com', |
| | keywords='computer vision, pytorch, image restoration, super-resolution, esrgan, real-esrgan', |
| | url='https://github.com/xinntao/Real-ESRGAN', |
| | include_package_data=True, |
| | packages=find_packages(exclude=('options', 'datasets', 'experiments', 'results', 'tb_logger', 'wandb')), |
| | classifiers=[ |
| | 'Development Status :: 4 - Beta', |
| | 'License :: OSI Approved :: Apache Software License', |
| | 'Operating System :: OS Independent', |
| | 'Programming Language :: Python :: 3', |
| | 'Programming Language :: Python :: 3.7', |
| | 'Programming Language :: Python :: 3.8', |
| | ], |
| | license='BSD-3-Clause License', |
| | setup_requires=['cython', 'numpy'], |
| | install_requires=get_requirements(), |
| | zip_safe=False) |
| |
|
