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-mm-entity-linking

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

from PIL import Image
from mmkg_mm_entity_linking import MMEntityLinking

image = Image.open("path/to/image")
probs = MMEntityLinking().inference(image, ["a dog", "a cat"])

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.

mmkg_mm_entity_linking-0.1.2-py3-none-any.whl (1.4 MB view details)

Uploaded Python 3

File details

Details for the file mmkg_mm_entity_linking-0.1.2-py3-none-any.whl.

File metadata

File hashes

Hashes for mmkg_mm_entity_linking-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 3cd017a066f718c912490ed7db991a286bd85aac635c2974281921a521534796
MD5 fa7e6b8b40d019f65f43fec14b14173e
BLAKE2b-256 ba39339c7eb074935e3dba531126da5d2ed0e085c3b10a2be6e997f0c2ae5c13

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