Metadata-Version: 2.1
Name: mmkg-visual-entity-recognition
Version: 0.2.5
Summary: MMKG Models
Home-page: https://github.com/vivym/mmkg-visual-entity-recognition
Download-URL: https://github.com/vivym/mmkg-visual-entity-recognition/tags
Author: Ming Yang
Author-email: ymviv@qq.com
Requires-Python: >=3.9
Description-Content-Type: text/markdown
Requires-Dist: pillow
Requires-Dist: aiofiles
Requires-Dist: fastapi
Requires-Dist: uvicorn[standard]
Requires-Dist: python-multipart

# 视觉实体识别使用说明

## 1. 环境依赖

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

```bash
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
```

## 2. 下载安装

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

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

```python
from PIL import Image
from mmkg_visual_entity_recognition import ImageEntityRecognition

image = Image.open("path/to/image")
res = ImageEntityRecognition.inference(image)
print(res)
```
