Metadata-Version: 2.1
Name: ThreeDJCG
Version: 0.1.1
Summary: Joint Captioning and Grounding
Home-page: https://github.com/ScaryJellyfish/ThreeDJCG
Author: Rui Zheng
Author-email: 1156486807@qq.com
License: UNKNOWN
Download-URL: https://github.com/ScaryJellyfish/ThreeDJCG/tags
Platform: UNKNOWN
Requires-Python: >=3.7
Description-Content-Type: text/markdown
Requires-Dist: pillow
Requires-Dist: aiofiles
Requires-Dist: fastapi
Requires-Dist: uvicorn[standard]
Requires-Dist: python-multipart
Requires-Dist: ftfy
Requires-Dist: regex
Requires-Dist: tqdm
Requires-Dist: h5py
Requires-Dist: easydict

# 视觉-文本实体链接使用说明

## 1. 环境依赖

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

```bash
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
```

## 2. 下载安装

可使用如下命令下载安装算法包：
```bash
pip install -U ThreeDJCG
```

## 3. 使用示例及运行参数说明

```python
from ThreeDJCG import ThreeDJCG

data_dict = {}
point_clouds = torch.ones([1, 10, 4]).cuda()
data_dict['point_clouds'] = point_clouds

data_dict["lang_num"] = 1
data_dict["lang_feat_list"] = torch.zeros([1, 1, 126, 300]).cuda()
data_dict["lang_len_list"] = torch.Tensor([10]).cuda()
data_dict["main_lang_feat_list"] = torch.zeros([1, 1, 126, 300]).cuda()
data_dict["main_lang_len_list"] = torch.Tensor([10]).cuda()
data_dict["first_obj_list"] = torch.Tensor([0]).cuda()
data_dict["unk_list"] = torch.zeros([1, 300]).cuda()
data_dict["unk"] = torch.zeros([1, 300]).cuda()
data_dict["istrain"] = torch.Tensor([0]).cuda()

data_dict = ThreeDJCG.inference(data_dict)
```


