Unsupervised Monocular Depth Estimation with Left-Right Consistency代码详解(model.py)

匿名 (未验证) 提交于 2019-12-03 00:02:01

定义MonodepthModel类
**def init**的第一个参数必须为delf

self.params = params self.mode = mode self.left = left self.right = right

self.model_collection已训练模型
reuse_variables=none

self.build_model() self.build_output()

如果是test模式,不进行build.losses和build_summaries

gradiebt_x(self,img)函数:猜测是计算img中x方向的梯度

upsample_nn(x,ratio)函数
:上采样,使用最近邻插值将X调整为【h* ratio,w*ratio】

scale_pyramid(img,num_scales) 没看太懂,大概在搭建图像金字塔

generate_image_left、generate_image_left 双线性采样(bilinear_sampler.py)

SSIM函数:定义SSIM运算方程

def get_disparity_smoothness(self, disp, pyramid):计算视差平滑度
disp_gradients_x、y:计算disp的x、y方向的偏导
image_gradients_x、y:计算img的x、y方向的偏导

disp = 0.3 * self.conv(x, 2, 3, 1, tf.nn.sigmoid):进行网络架构搭建

网络架构

def conv(self, x, num_out_layers, kernel_size, stride, activation_fn=tf.nn.elu):

默认激励函数:ELU

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!