English | SNE-RoadSeg2 介绍 此SNE-RoadSeg2基于SNE-RoadSeg的官方pytorch实现的,已被接受。这是他们的。 在此,我们为提供了培训和测试设置。我们在Python 3.7,CUDA 10.0,cuDNN 7和PyTorch 1.1中测试我们的代码。我们提供Dockerfile来构建我们使用的Dockerfile映像。 设置 请根据以下文件夹结构设置KITTI道路数据集和预训练的权重: SNE-RoadSeg |-- checkpoints | |-- kitti | | |-- kitti_net_RoadSeg.pth |-- data |-- datasets | |-- kitti | | |-- training | | | |-- calib | | | |-- depth_u16 |
1