Skip to main content

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)

Project details


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

If you're not sure about the file name format, learn more about wheel file names.

File details

Details for the file mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl.

File metadata

File hashes

Hashes for mmkg_visual_entity_recognition-0.2.6-py3-none-any.whl
Algorithm Hash digest
SHA256 3d5328eeb6da99d8441f89e9ea59e419061bbb9c34274d01be1aeeddac88d3ce
MD5 c7ca36d6dd795bcf0ace8384a8897053
BLAKE2b-256 c82f03a78f3e47e440b93077b61f5a96aad7f220a87e9410b7c07a7e37c3f898

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page