RTFNet:RGB热融合网络用于城市场景的语义分割

RTFNet-火炬 这是RTFNet的官方pytorch实施（IEEE RAL）。 一些代码是从借用的。 请注意，我们评估指标（Acc和IoU）的实现与MFNet中的实现不同。 此外，我们在计算指标时会考虑未标记的类。 当前版本支持Python> = 3.6，CUDA> = 11.1和PyTorch> = 1.7.1，但是它在较低版本的CUDA和PyTorch上应该可以正常工作。 请根据需要修改Dockerfile 。 如果您不使用docker，请手动安装Dockerfile列出的依赖Dockerfile 。 介绍 RTFNet是用于语义分割的数据融合网络。 它由两个编码器和一个解码器组成。 尽管RTFNet是用RGB热数据设计的，但它对RGB-D数据的推广很好。 请看一下。 数据集 原始数据集可以从MFNet项目下载，但建议您从下载经过预处理的数据集。 预先训练的体重 本文中使用的权重