Skip to main content

MMKG Models

Project description

场景图理解使用说明

1. 算法描述

2. 依赖及安装

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

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

可使用如下命令下载安装算法包:

pip install -U mmkg-sgg

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

from mmkg_sgg import SGG
image_path = 'test.jpg'
meta = {'image_size': (224, 224)}
output = SGG().inference(image_path, meta)

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

mmkg_sgg-0.1.4-cp37-cp37m-manylinux1_x86_64.whl (5.5 MB view hashes)

Uploaded CPython 3.7m

Supported by

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