Metadata-Version: 2.1
Name: scene-graph-predictor-pc
Version: 0.1.1
Summary: 3D Scene Graph Models
Home-page: https://github.com/wtt0213/scene-graph-predictor-pc
Download-URL: https://github.com/wtt0213/scene-graph-predictor-pc/tags
Author: Ziqin Wang
Author-email: wzqbuaa@qq.com
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: trimesh (==3.12.7)
Requires-Dist: open3d (==0.16.0)
Requires-Dist: aiofiles
Requires-Dist: fastapi
Requires-Dist: uvicorn[standard]
Requires-Dist: python-multipart

# 点云场景图预测使用说明

# 1. 环境依赖

CUDA版本: 11.3
其他依赖库的安装命令如下：

```bash
conda create -n py38 python=3.8
```

```bash
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html
pip install torch-geometric
```

# 2. 下载安装

```bash
pip install scene-graph-predictor-pc
```

# 3. 使用说明
```python
import scene_graph_predictor_pc
import trimesh

plydata=trimesh.load('scene/scene0/labels.instances.align.annotated.v2.ply', process=False)
model = scene_graph_predictor_pc.SceneGraphPredictor()
# model 加载数据
model.load('checkpoint')
# 模型预测

```
