文件名称：object_relation_transformer:用于图像字幕的对象关系转换器的实现
文件大小：955KB
文件格式：ZIP
更新时间：2024-06-15 15:20:59
machine-learning Python
对象关系变压器 这是在NeurIPS 2019中发布的对象关系变压器的PyTorch实现。您可以在找到本文。 该存储库主要基于罗若田针对图像字幕GitHub repo进行的自关键序列训练的代码，可在找到。 主要添加内容如下： 关系变压器模型 用于创建在MSCOCO上运行的报告的脚本 要求 Python 2.7（因为Python 3没有版本） PyTorch 0.4+（以及Torchvision） h5py scikit图像 打字 pyemd Gensim （已作为子模块添加）。 请参阅.gitmodules并将引用的.gitmodules库克隆到object_relation_transformer文件夹中。 库，用于生成不同的评估指标。 要进行设置，请将存储库克隆到object_relation_transformer文件夹中。 确保将克隆的存储库文件夹名称保留为coco
【文件预览】：
object_relation_transformer-master
----dataloaderraw.py(5KB)
----neurips_report_commands.sh(5KB)
----train.py(11KB)
----.gitmodules(81B)
----models()
--------RelationTransformerModel.py(25KB)
--------ShowTellModel.py(8KB)
--------TransformerModel.py(23KB)
--------CaptionModel.py(9KB)
--------AttModel.py(30KB)
--------__init__.py(2KB)
--------FCModel.py(8KB)
--------OldModel.py(11KB)
--------AttEnsemble.py(13KB)
----dataloader.py(13KB)
----opts.py(11KB)
----scripts()
--------prepro_labels.py(8KB)
--------prepro_feats.py(4KB)
--------prepro_ngrams.py(5KB)
--------prepro_bbox_relative_coords.py(4KB)
--------make_bu_data.py(2KB)
--------copy_model.sh(117B)
----neurips_training_runs.sh(7KB)
----media()
--------Object Relation_Transformer_Poster.pdf(1.02MB)
----eval_utils.py(7KB)
----vis()
--------index.html(2KB)
--------imgs()
--------jquery-1.8.3.min.js(91KB)
----LICENSE(1KB)
----create_report.py(3KB)
----eval_ensemble.py(7KB)
----eval.py(8KB)
----README.md(15KB)
----Code-of-Conduct.md(7KB)
----Contributing.md(2KB)
----misc()
--------resnet_utils.py(698B)
--------rewards.py(3KB)
--------report.py(29KB)
--------correct_coco_eval_cap.py(1KB)
--------utils.py(14KB)
--------__init__.py(0B)
--------resnet.py(2KB)
----.gitignore(224B)
----LICENSE-CC-BY-4.md(14KB)