MMKG Models
Project description
视觉实体识别使用说明
1. 环境依赖
CUDA版本: 11.7 其他依赖库的安装命令如下：
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
2. 下载安装
可使用如下命令下载安装算法包：
pip install -U mmkg-visual-entity-recognition
3. 使用示例及运行参数说明
from PIL import Image
from mmkg_visual_entity_recognition import ImageEntityRecognition
image = Image.open("path/to/image")
label_id, label, prob = ImageEntityRecognition().inference(image)
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
File details
Details for the file mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl.
File metadata
- Download URL: mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl
- Upload date:
- Size: 13.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.9.16
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 3d5328eeb6da99d8441f89e9ea59e419061bbb9c34274d01be1aeeddac88d3ce |
|
| MD5 | c7ca36d6dd795bcf0ace8384a8897053 |
|
| BLAKE2b-256 | c82f03a78f3e47e440b93077b61f5a96aad7f220a87e9410b7c07a7e37c3f898 |
