Scene-Graph-Benchmark.pytorch 项目推荐

Scene-Graph-Benchmark.pytorch 项目推荐

Scene-Graph-Benchmark.pytorch A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020” 项目地址: https://gitcode.com/gh_mirrors/sc/Scene-Graph-Benchmark.pytorch

项目基础介绍

Scene-Graph-Benchmark.pytorch 是一个基于 PyTorch 的场景图生成开源项目，由 KaihuaTang 开发并维护。该项目旨在构建一个新的场景图生成（SGG）代码库，同时也是论文《Unbiased Scene Graph Generation from Biased Training》的 PyTorch 实现。

编程语言

项目主要使用 Python 语言，基于 PyTorch 深度学习框架。

核心功能

• 提供了场景图生成的标准协议，包括谓词分类（PredCls）、场景图分类（SGCls）和场景图检测（SGDet）。
• 利用最新的 Faster R-CNN 进行预训练，并在 reimplemented VCTree 基础上达到新的 state-of-the-art Recall@k。
• 支持自定义图像上的场景图检测和可视化。
• 提供了详细的指标解释和结果报告，帮助用户更好地理解和评估模型性能。

最近更新的功能

• 增加了无图约束的平均召回率（ng-mR@K）和无图约束的零样本召回率（ng-zR@K）。
• 允许在自定义图像上进行场景图检测。
• 将自定义图像的场景图检测结果输出格式更改为 json 文件。
• 提供了自定义图像场景图检测结果的可视化功能。

Scene-Graph-Benchmark.pytorch A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020” 项目地址: https://gitcode.com/gh_mirrors/sc/Scene-Graph-Benchmark.pytorch