【文件属性】:
文件名称:energy-based-scene-graph:场景图生成的基于能量的学习的代码发布
文件大小:27.77MB
文件格式:ZIP
更新时间:2021-05-20 10:18:44
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