视觉实体识别使用说明
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)
1 maintainer
Release files for mmkg-visual-entity-recognition 0.2.6
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl | Python 3 | none | any | Details |
Release files / mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl
| Download URL | mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl |
|---|---|
| Size | 13.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
3d5328eeb6da99d8441f89e9ea59e419061bbb9c34274d01be1aeeddac88d3ce
|
|
BLAKE2b-256 checksum How to use checksums |
c82f03a78f3e47e440b93077b61f5a96aad7f220a87e9410b7c07a7e37c3f898
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.9.16
|
