diff --git a/README.md b/README.md index 5a3578e..4276f02 100644 --- a/README.md +++ b/README.md @@ -73,6 +73,14 @@ pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https pip install -r requirements.txt pip install flash-attn --no-build-isolation ``` +or +```shell +pip install git+https://github.com/buaacyw/MeshAnything.git +``` +And directly use in your code as +``` +import MeshAnything +``` ## Usage ### Local Gradio Demo @@ -100,7 +108,7 @@ python main.py --input_dir examples --out_dir mesh_output --input_type mesh --mc python main.py --input_dir pc_examples --out_dir pc_output --input_type pc_normal # inference for single file -python main.py --input_dir pc_examples/mouse.npy --out_dir pc_output --input_type pc_normal +python main.py --input_path pc_examples/mouse.npy --out_dir pc_output --input_type pc_normal ``` ## Important Notes diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..7c63b3b --- /dev/null +++ b/setup.py @@ -0,0 +1,32 @@ +from pathlib import Path +from setuptools import setup, find_packages + +setup_path = Path(__file__).parent +README = (setup_path / "README.md").read_text(encoding="utf-8") + +with open("README.md", "r") as fh: + long_description = fh.read() + +def split_requirements(requirements): + install_requires = [] + dependency_links = [] + for requirement in requirements: + if requirement.startswith("git+"): + dependency_links.append(requirement) + else: + install_requires.append(requirement) + + return install_requires, dependency_links + +with open("./requirements.txt", "r") as f: + requirements = f.read().splitlines() + +install_requires, dependency_links = split_requirements(requirements) + +setup( + name = "MeshAnything", + packages=find_packages(), + description=long_description, + long_description=README, + install_requires=install_requires +)