PSM-Net根据Stereo图像生成depth图像

一、新建文件夹

在KITTI数据集下新建depth_0目录

二、激活anaconda环境

conda activate pt14py37

三、修改submission.py文件

3.1 KITTI数据集路径

parser.add_argument('--datapath', 
default='/home/njust/KITTI_DataSet/00/',
help='select model')

3.2 深度图像输出路径

save_path = '/home/njust/KITTI_DataSet/00/depth_0/'

四、进行推理

python submission.py

你永远不知道自己的缺点有多少，去询问朋友的建议或者自我反思。