论文标题
多任务gan,用于语义细分和深度完成,并具有循环一致性
Multi-task GANs for Semantic Segmentation and Depth Completion with Cycle Consistency
论文作者
论文摘要
语义细分和深度完成是场景理解中的两个具有挑战性的任务,它们被广泛用于机器人技术和自动驾驶中。尽管建议使用一些小的修改(例如更改最后一层)共同训练这两个任务,但一项任务的结果并未用于改善另一个任务的结果,尽管这两个任务之间存在一些相似之处。在本文中,我们提出了多任务生成对抗网络(多任务gan),它们不仅在语义细分和深度完成方面都有能力,而且还通过生成的语义图像提高了深度完成的准确性。此外,我们通过引入多尺度的空间池块和结构相似性重建损失来改善基于Cyclegan生成的语义图像的细节。此外,考虑到语义结构和几何结构之间的内部一致性,我们开发出语义引导的平滑度损失,以改善深度完成结果。关于CityScapes数据集和Kitti深度完成基准的广泛实验表明,多任务gans能够为语义细分和深度完成任务实现竞争性能。
Semantic segmentation and depth completion are two challenging tasks in scene understanding, and they are widely used in robotics and autonomous driving. Although several works are proposed to jointly train these two tasks using some small modifications, like changing the last layer, the result of one task is not utilized to improve the performance of the other one despite that there are some similarities between these two tasks. In this paper, we propose multi-task generative adversarial networks (Multi-task GANs), which are not only competent in semantic segmentation and depth completion, but also improve the accuracy of depth completion through generated semantic images. In addition, we improve the details of generated semantic images based on CycleGAN by introducing multi-scale spatial pooling blocks and the structural similarity reconstruction loss. Furthermore, considering the inner consistency between semantic and geometric structures, we develop a semantic-guided smoothness loss to improve depth completion results. Extensive experiments on Cityscapes dataset and KITTI depth completion benchmark show that the Multi-task GANs are capable of achieving competitive performance for both semantic segmentation and depth completion tasks.