if __name__ == "__main__": file_path = opt.data_path file_list = opt.test_list f = open(file_list, 'r') filelist = f.readlines() for index in range(len(filelist)): current_file = filelist[index] if opt.kitti2015: leftname = file_path + 'image_2/' + current_file[0: len(current_file) - 1] rightname = file_path + 'image_3/' + current_file[0: len(current_file) - 1] if opt.kitti: leftname = file_path + 'colored_0/' + current_file[0: len(current_file) - 1] rightname = file_path + 'colored_1/' + current_file[0: len(current_file) - 1] savename = opt.save_path + current_file[0: len(current_file) - 1] test(leftname, rightname, savename)
时间: 2024-03-04 16:51:34 浏览: 57
graph_opt.pb
这段代码是程序的入口,首先读取文件路径和测试文件列表,然后逐行读取测试文件列表中的文件,并根据文件路径和文件名拼接出左右两张图片的完整路径。如果是 kitti2015 数据集,左右两张图片的路径分别在 image_2 和 image_3 文件夹下;如果是 kitti 数据集,则左右两张图片的路径分别在 colored_0 和 colored_1 文件夹下。然后调用 test 函数对左右两张图片进行处理,得到深度图并保存在指定路径下。这个函数的输入参数包括读取的文件路径,测试文件列表,保存深度图的路径和其他一些参数,返回值是 None。
阅读全文