honeymoose
/
MeshAnything
mirror of https://github.com/buaacyw/MeshAnything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/buaacyw/MeshAnything

This commit is contained in:
Yiwen Chen 2024-08-05 12:57:19 +08:00
parent f5b3356d62 d53e598283
commit 362c99f31f
2 changed files with 43 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
README.md
View File

@ -67,12 +67,20 @@ Our environment has been tested on Ubuntu 22, CUDA 11.8 with A100, A800 and A600
1. Clone our repo and create conda environment 1. Clone our repo and create conda environment
``` ```
git clone https://github.com/buaacyw/MeshAnything.git && cd MeshAnything git clone https://github.com/buaacyw/MeshAnything.git && cd MeshAnything
conda create -n MeshAnything python==3.10.13 conda create -n MeshAnything python==3.10.13 -y
conda activate MeshAnything conda activate MeshAnything
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118 pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt pip install -r requirements.txt
pip install flash-attn --no-build-isolation 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 ## Usage
### Local Gradio Demo <a href='https://github.com/gradio-app/gradio'><img src='https://img.shields.io/github/stars/gradio-app/gradio'></a> ### Local Gradio Demo <a href='https://github.com/gradio-app/gradio'><img src='https://img.shields.io/github/stars/gradio-app/gradio'></a>
@ -86,7 +94,7 @@ python app.py
python main.py --input_dir examples --out_dir mesh_output --input_type mesh python main.py --input_dir examples --out_dir mesh_output --input_type mesh
# single file input # single file input
python main.py --input_path examples/wand.ply --out_dir mesh_output --input_type mesh python main.py --input_path examples/wand.obj --out_dir mesh_output --input_type mesh
# Preprocess with Marching Cubes first # Preprocess with Marching Cubes first
python main.py --input_dir examples --out_dir mesh_output --input_type mesh --mc python main.py --input_dir examples --out_dir mesh_output --input_type mesh --mc
@ -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 python main.py --input_dir pc_examples --out_dir pc_output --input_type pc_normal
# inference for single file # 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 ## Important Notes

32
setup.py Normal file
View File

@ -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
)