文件名称：energy-based-scene-graph:场景图生成的基于能量的学习的代码发布
文件大小：27.77MB
文件格式：ZIP
更新时间：2024-06-09 04:05:24
JupyterNotebook
基于能量的场景图生成学习 该存储库包含CVPR 2021接受的论文《的代码。 环境设定 要设置具有所有必需依赖关系的环境，请遵循的步骤。 注意：默认情况下， cudatoolkit版本设置为10.0。 在计算机上创建环境时，请通过运行nvcc --version检查cuda编译器版本，并适当调整cudatoolkit版本。 运行代码时，版本不匹配会导致build失败或segmentaion fault错误。 数据集 检查以获取有关下载数据集的详细信息。 预训练模型 我们在使用基于交叉熵和基于能量的训练方法训练的Visual Genome数据集上实现了保留的VCTree模型的权重。 循证医学 CE认证 要训练您自己的模型，您可以从此获取预训练探棒的权重。 基于能量的场景图生成的培训 python -m torch.distributed.launch --master_port 10