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