mmkg-sgg 0.1.5

MMKG Models

图像场景图生成使用说明

1. 算法描述

该算法用于图像场景图生成(SGG)。该算法基于PyTorch框架开发，输入一张图片，算法会检测图片中人和物的交互关系，输出图像场景图。

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)