Spaces:
Running
Running
| import os | |
| import subprocess | |
| import time | |
| from setuptools import find_packages, setup | |
| os.system('python -m pip install Openmim') | |
| os.system('python -m mim install mmengine') | |
| os.system('python -m mim install "mmcv-full==1.6.2"') | |
| os.system('python -m mim install "mmpose==0.29.0"') | |
| os.system('python -m mim install "gradio==3.44.0"') | |
| def readme(): | |
| with open('README.md', encoding='utf-8') as f: | |
| content = f.read() | |
| return content | |
| version_file = 'models/version.py' | |
| def get_git_hash(): | |
| def _minimal_ext_cmd(cmd): | |
| # construct minimal environment | |
| env = {} | |
| for k in ['SYSTEMROOT', 'PATH', 'HOME']: | |
| v = os.environ.get(k) | |
| if v is not None: | |
| env[k] = v | |
| # LANGUAGE is used on win32 | |
| 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] | |
| elif os.path.exists(version_file): | |
| try: | |
| from models.version import __version__ | |
| sha = __version__.split('+')[-1] | |
| except ImportError: | |
| raise ImportError('Unable to get git version') | |
| else: | |
| sha = 'unknown' | |
| return sha | |
| def write_version_py(): | |
| content = """# GENERATED VERSION FILE | |
| # TIME: {} | |
| __version__ = '{}' | |
| short_version = '{}' | |
| version_info = ({}) | |
| """ | |
| sha = get_hash() | |
| with open('models/VERSION', 'r') as f: | |
| SHORT_VERSION = f.read().strip() | |
| VERSION_INFO = ', '.join(SHORT_VERSION.split('.')) | |
| VERSION = SHORT_VERSION + '+' + sha | |
| version_file_str = content.format(time.asctime(), VERSION, SHORT_VERSION, | |
| 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='pose_anything', | |
| version=get_version(), | |
| description='A template for pytorch projects.', | |
| long_description=readme(), | |
| packages=find_packages(exclude=('configs', 'tools', 'demo')), | |
| package_data={'pose_anything.ops': ['*/*.so']}, | |
| classifiers=[ | |
| 'Development Status :: 4 - Beta', | |
| 'License :: OSI Approved :: Apache Software License', | |
| 'Operating System :: OS Independent', | |
| 'Programming Language :: Python :: 3', | |
| 'Programming Language :: Python :: 3.5', | |
| 'Programming Language :: Python :: 3.6', | |
| 'Programming Language :: Python :: 3.7', | |
| ], | |
| license='Apache License 2.0', | |
| setup_requires=['pytest-runner', 'cython', 'numpy'], | |
| tests_require=['pytest', 'xdoctest'], | |
| install_requires=get_requirements(), | |
| zip_safe=False) | |