多恩 更新 更新了整个代码库,并重新实现了一些层和损失函数,以使其运行速度更快并使用更少的内存。 该存储库仅包含 DORN 模型的核心代码。 整个代码将保存在。 介绍 这是的的 PyTorch 实现。 预训练模型 DORN 的 resnet 主干,在第一 conv 层有 3 个 conv,与原始 resnet 不同。 resnet骨干网的预训练模型可以从下载 。 数据集 纽约大学深度 V2 未实现。 基蒂 根据 ,我们应该远离 eigen split 并切换到 。 更多详细信息,请参阅 。
2021-10-26 15:22:36 13KB pytorch ordinal-regression dorn depth-prediction
1