monodepth代码与原理对照实现

赛亚超2024-07-13 21:38

先实现demomonodepth/monodepth_simple.py at master · mrharicot/monodepth · GitHub

复制代码

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='0'

这行代码是为tensorflow设置环境变量TF_CPP_MIN_LOG_LEVEL,用来控制tensorflow c++后端输出的日志级别。=0就是输出所有级别的日志信息。包括(调试信息)DEBUG,信息（INFO），警告（WARNING）,错误（ERROR）和致命错误（FATAL）。这是用于调试作为目的，提供大量的输出信息。

添加命令行参数，其中包括编码器resnet50，图像路径image_path,权重文件checkpoint_path,输入图像的长宽大小。

复制代码

parser.add_argument('--encoder',          type=str,   help='type of encoder, vgg or resnet50', default='vgg')
parser.add_argument('--image_path',       type=str,   help='path to the image', required=True)
parser.add_argument('--checkpoint_path',  type=str,   help='path to a specific checkpoint to load', required=True)
parser.add_argument('--input_height',     type=int,   help='input height', default=256)
parser.add_argument('--input_width',      type=int,   help='input width', default=512)

下一步就是视差的函数创建，这一部分还不太了解，最好是论文的学习。