mtan:实施“注意力集中的端到端多任务学习” [CVPR 2019]-源码

MTAN-多任务注意力网络 该存储库包含多任务注意力网络（MTAN）的源代码，以及来自Shikun ， 和引入基线。 请参阅我们的项目页面的详细结果。 实验 图像到图像预测（一对多） 在文件夹im2im_pred ，我们提供了建议的网络以及本文介绍的NYUv2数据集上的所有基线。 所有模型都是用PyTorch编写的，并且在最新的提交中，我们已将实现更新为PyTorch 1.5版。 下载我们经过预处理的NYUv2数据集。 我们从使用预先计算的地面真实法线。 原始的13类NYUv2数据集可以使用定义的分段标签直接下载到。 很抱歉，由于意外的计算机崩溃，我无法提供原始的预处理代码。 更新-2019年6月：我现在发布了具有2、7和19类语义标签（请参阅本文以获取更多详细信息）和（反）深度标签的预处理CityScapes数据集。 下载[256×512个，2.42GB]版本 和[128×2