File size: 1,777 Bytes
06e9d12 |
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 43 44 45 46 47 48 49 50 51 |
#!/usr/bin/env python
import subprocess
import os
import pkg_resources
from setuptools import setup, find_packages
ProjectDir = os.path.dirname(__file__)
result = subprocess.run(["pip", "install", "basicsr"], capture_output=True, text=True)
result = subprocess.run(
["pip", "install", "--no-cache-dir", "-U", "openmim"],
capture_output=True,
text=True,
)
result = subprocess.run(["mim", "install", "mmengine"], capture_output=True, text=True)
result = subprocess.run(
["mim", "install", "mmcv>=2.0.1"], capture_output=True, text=True
)
result = subprocess.run(
["mim", "install", "mmdet>=3.1.0"], capture_output=True, text=True
)
result = subprocess.run(
["mim", "install", "mmpose>=1.1.0"], capture_output=True, text=True
)
with open(os.path.join(ProjectDir, "requirements.txt"), "r") as f:
requirements = f.read().splitlines()
requirements = [x for x in requirements if x and not x.startswith("#")]
requirements = [x.split(" ")[0] if "index-url" in x else x for x in requirements]
setup(
name="musev", # used in pip install
version="1.0.0",
author="anchorxia, zkangchen",
author_email="anchorxia@tencent.com, zkangchen@tencent.com",
description="Package about human video creation",
# long_description=long_description,
# long_description_content_type="text/markdown",
url="https://github.com/TMElyralab/MuseV",
packages=find_packages("musev"),
package_dir={"": "musev"},
# include_package_data=True, # please edit MANIFEST.in
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
install_requires=requirements,
# dependency_links=["https://download.pytorch.org/whl/cu118],
)
|